Update app.py

app.py

@@ -195,7 +195,7 @@ def infer():
     flow_img = flow_to_image(predicted_flow).to("cpu")
 #     output_folder = "/tmp/"  # Update this to the folder of your choice
     write_jpeg(flow_img, f"predicted_flow.jpg")
-    input_image = flow_to_image(img1_batch).to("cpu")
+    input_image = flow_to_image(frames[100]).to("cpu")
     write_jpeg(input_image, f"frame_input.jpg")
     flo_file = write_flo(predicted_flow, "flofile.flo")
     #write_jpeg(frames[100], f"input_image.jpg")