Update app.py

app.py

@@ -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], predicted_flow, blend=0.5)
+    warp_res = warp(img1_batc, img2_batc, predicted_flow, blend=0.5)
     return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
 ####################################
 # Bonus: Creating GIFs of predicted flows