Update app.py

app.py

@@ -147,7 +147,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")
-    flo_file = write_flo(predicted_flow, "flofile.flo").to("cpu")
+    flo_file = write_flo(predicted_flow, "flofile.flo")
     return "done", "predicted_flow.jpg", flo_file
 ####################################
 # Bonus: Creating GIFs of predicted flows