Update app.py

app.py

@@ -63,11 +63,31 @@ def warp_flow(img, flow):
       res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
       return res
 
-  
+def read_flo_file(filename):
+    """
+    Read from Middlebury .flo file
+    :param flow_file: name of the flow file
+    :return: optical flow data in matrix
+    """
+    f = open(filename, 'rb')
+    magic = np.fromfile(f, np.float32, count=1)
+    data2d = None
+
+    if 202021.25 != magic:
+        print('Magic number incorrect. Invalid .flo file')
+    else:
+        w = np.fromfile(f, np.int32, count=1)
+        h = np.fromfile(f, np.int32, count=1)
+        print("Reading %d x %d flow file in .flo format" % (h, w))
+        data2d = np.fromfile(f, np.float32, count=2 * w * h)
+        # reshape data into 3D array (columns, rows, channels)
+        data2d = np.resize(data2d, (h[0], w[0], 2))
+    f.close()
+    return data2d
 
 def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
     #flo_path = flo_path.cpu().data.numpy()
-    flow21 = np.load(flo_path, allow_pickle=True)
+    flow21 = read_flo_file(flo_path)
     frame1pil = np.array(frame1.convert('RGB').resize((flow21.shape[1],flow21.shape[0])))
     frame1_warped21 = warp_flow(frame1pil, flow21)
     # frame2pil = frame1pil