Update app.py

app.py

@@ -66,7 +66,7 @@ def warp_flow(img, flow):
   
 
 def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
-    flo_path = flo_path.cpu().data.numpy()
+    #flo_path = flo_path.cpu().data.numpy()
     flow21 = np.load(flo_path, allow_pickle=True)
     frame1pil = np.array(frame1.convert('RGB').resize((flow21.shape[1],flow21.shape[0])))
     frame1_warped21 = warp_flow(frame1pil, flow21)
@@ -175,7 +175,7 @@ def infer():
     write_jpeg(flow_img, f"predicted_flow.jpg")
     flo_file = write_flo(predicted_flow, "flofile.flo")
                         
-    warp_res = warp(frames[100], frames[101], predicted_flow, blend=0.5)
+    warp_res = warp(frames[100], frames[101], "flofile.flo", blend=0.5)
     return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
 ####################################
 # Bonus: Creating GIFs of predicted flows