Update app.py

app.py

@@ -121,9 +121,9 @@ def infer():
     predicted_flow = list_of_flows[-1][0]
     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_{i}.jpg")
+    write_jpeg(flow_img, f"predicted_flow.jpg")
 
-    return "done", "predicted_flow_{i}.jpg"
+    return "done", "predicted_flow.jpg"
 ####################################
 # Bonus: Creating GIFs of predicted flows
 # ---------------------------------------