Update app.py

app.py

@@ -211,7 +211,7 @@ def infer():
     print(f"img1_batch dtype: {img1_batch.dtype}")
     #res = get_warp_res(img1_batch, "predicted_flow.jpg", 'warped.png')
 
-    numpy_array_flow = predicted_flow.numpy()
+    numpy_array_flow = predicted_flow.permute(1, 2, 0).detach().cpu().numpy()
     print(f"numpy_array_flow: {numpy_array_flow}")
     print(f"numpy_array_flow shape: {numpy_array_flow.shape}")
     print(f"numpy_array_flow dtype: {numpy_array_flow.dtype}")