Update app.py

app.py

@@ -61,7 +61,7 @@ def write_flo(flow, filename):
 #warp using scipy
 def warp_image(im, flow):
     print(f"IMAGE: {im}")
-    im = im.cpu().data.numpy()
+    im = im[0].cpu().data.numpy()
     im = im.astype(np.float32)
     print(f"IMAGE AP: {im}")
     print(f"FLOW AV: {flow}")