Update app.py

app.py

@@ -196,7 +196,7 @@ def infer():
     write_jpeg(flow_img, f"predicted_flow.jpg")
     flo_file = write_flo(predicted_flow, "flofile.flo")
                         
-    warp_res = warp(img1_batch, img2_batch, predicted_flow, blend=0.5)
+    warp_res = warp(img1_batch, img2_batch, "predicted_flow.jpg", blend=0.5)
     return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
 ####################################
 # Bonus: Creating GIFs of predicted flows