Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 4, 2023

Commit

a570d5d

1 Parent(s): 8084825

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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