Update app.py

app.py

@@ -211,8 +211,8 @@ def infer():
     print(f"img1_batch dtype: {img1_batch.dtype}")
     #res = get_warp_res(img1_batch, "predicted_flow.jpg", 'warped.png')
 
-    h, w = predicted_flows.shape[:2]
-    flow = predicted_flows.copy()
+    h, w = predicted_flow.shape[:2]
+    flow = predicted_flow.copy()
     flow[:, :, 0] + np.arange(w)
     flow[:, :, 1] + np.arange(h)[:, np.newaxis]
     # print('flow stats', flow.max(), flow.min(), flow.mean())