Update app.py

app.py

@@ -513,6 +513,12 @@ def main(input_path, model_path, output_dir, need_animation=False, resize_h=None
                save_all=True, append_images=all_frames[1:], optimize=False, duration=40, loop=0)
         return os.path.join(frame_dir, "animation.gif"), tensor_to_pil
 
+def load_image( infilename ) :
+    img = Image.open( infilename )
+    img.load()
+    data = np.asarray( img, dtype="int32" )
+    return data
+    
 def gradio_inference(image):
     return main(input_path=image.name,
          model_path='model.pth',