Update app.py

app.py

@@ -63,12 +63,17 @@ def warp_image(im, flow):
     im = [(img1 + 1) / 2 for img1 in im]
     
     print(f"IMAGE: {im}")
-    im = im[0].cpu().data.numpy()
-    im = im.astype(np.float32)
+    #im = im[0].cpu().data.numpy()
+    #im = im.astype(np.float32)
+    im = torch.FloatTensor(im).permute(0, 3, 1, 2).contiguous()
+    
+    
     print(f"IMAGE AP: {im}")
     print(f"FLOW AV: {flow}")
-    flow = flow.cpu().data.numpy()
-    flow = flow.astype(np.float32)
+    #flow = flow.cpu().data.numpy()
+    #flow = flow.astype(np.float32)
+    flow = torch.FloatTensor(flow)
+    
     print(f"FLOW AP: {flow}")
     h, w = flow.shape[:2]
     flow[:,:,0] += np.arange(w)