Update app.py

app.py

@@ -30,13 +30,19 @@ import tempfile
 from pathlib import Path
 from urllib.request import urlretrieve
 
-def write_flo(filename, flow):
+def write_flo_file(flow, filename):
     """
-    write optical flow in Middlebury .flo format
+    Write optical flow in Middlebury .flo format
+    
     :param flow: optical flow map
     :param filename: optical flow file path to be saved
     :return: None
+    
+    from https://github.com/liruoteng/OpticalFlowToolkit/
+    
     """
+    # forcing conversion to float32 precision
+    flow = flow.astype(np.float32)
     f = open(filename, 'wb')
     magic = np.array([202021.25], dtype=np.float32)
     (height, width) = flow.shape[0:2]
@@ -46,7 +52,7 @@ def write_flo(filename, flow):
     w.tofile(f)
     h.tofile(f)
     flow.tofile(f)
-    f.close() 
+    f.close()
 
 
 def infer():
@@ -141,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("flofile.flo", predicted_flow).to("cpu")
+    flo_file = write_flo(predicted_flow, "flofile.flo").to("cpu")
     return "done", "predicted_flow.jpg", flo_file
 ####################################
 # Bonus: Creating GIFs of predicted flows