Update app.py

app.py

@@ -87,7 +87,7 @@ def read_flo_file(filename):
 
 def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
     #flo_path = flo_path.cpu().data.numpy()
-    flow21 = read_flo_file(flo_path)
+    flow21 = flo_path
     frame1pil = np.array(frame1.convert('RGB').resize((flow21.shape[1],flow21.shape[0])))
     frame1_warped21 = warp_flow(frame1pil, flow21)
     # frame2pil = frame1pil
@@ -195,7 +195,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], "flofile.flo", blend=0.5)
+    warp_res = warp(frames[100], frames[101], predicted_flow, blend=0.5)
     return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
 ####################################
 # Bonus: Creating GIFs of predicted flows