Update app.py

app.py

@@ -66,7 +66,8 @@ def write_flo(flow, filename):
 
 def warp_flow(img, flow, mul=1.):
     #img = np.array(img.convert('RGB'))  
-    flow = np.load(flow)
+    img = cv2.imread(img,mode='RGB')
+    #flow = np.load(flow)
     h, w = flow.shape[:2]
     flow = flow.copy()
     flow[:, :, 0] += np.arange(w)
@@ -82,11 +83,11 @@ def warp_flow(img, flow, mul=1.):
 
 def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
     print(f"FNAME IMAGE: {fname_image}")
-    im2 = imread(fname_image)
-    print(f"FNAME IMAGE READED: {im2.shape}")
+    #im2 = imread(fname_image)
+    #print(f"FNAME IMAGE READED: {im2.shape}")
     flow = fname_flow.cpu().detach().numpy()
     print(f"FNAME FLOW READED: {flow.shape}")
-    res = warp_flow(im2, flow, 1.)
+    res = warp_flow(fname_image, flow, 1.)
     
 def infer():
     video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"