Update app.py

app.py

@@ -104,7 +104,7 @@ def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
     print(f"FNAME IMAGE: {fname_image}")
     im2 = imread(fname_image)
     print(f"FNAME IMAGE READED: {im2}")
-    flow = fname_flow.cpu()
+    flow = fname_flow.cpu().detach().numpy()
     im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
     imwrite(fname_output, im_warped)