Spaces:

fffiloni
/

RAFT

Paused

fffiloni commited on Mar 4, 2023

Commit

7a81acf

1 Parent(s): bb9568b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,7 +42,7 @@ def write_flo(flow, filename):
     """
     # forcing conversion to float32 precision
-    flow.numpy().to("cpu")
     flow = flow.astype(np.float32)
     f = open(filename, 'wb')
     magic = np.array([202021.25], dtype=np.float32)

     """
     # forcing conversion to float32 precision
+    flow.cpu().data.numpy()
     flow = flow.astype(np.float32)
     f = open(filename, 'wb')
     magic = np.array([202021.25], dtype=np.float32)