Update app.py

app.py

@@ -193,7 +193,7 @@ def infer():
 #     output_folder = "/tmp/"  # Update this to the folder of your choice
     write_jpeg(flow_img, f"predicted_flow.jpg")
     flo_file = write_flo(predicted_flow, "flofile.flo")
-    write_jpeg(img1_batch, f"input_image.jpg")
+    write_jpeg(frames[100], f"input_image.jpg")
     #res = warp_image(img1_batch, predicted_flow)
     res = get_warp_res("input_image.jpg", "flofile.flo", fname_output='warped.png')
     print(res)