Update app.py

app.py

@@ -98,10 +98,12 @@ def warpImage(im, vx, vy, cast_uint8=True):
 
 
 def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
+    print(f"FNAME IMAGE: {fname_image")
     im2 = imread(fname_image)
-    flow = readFlowFile(fname_flow)
-    im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
-    imwrite(fname_output, im_warped)
+    print(f"FNAME IMAGE READED: {im2}")
+    #flow = readFlowFile(fname_flow)
+    #im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
+    #imwrite(fname_output, im_warped)
     
 def infer():
     video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"
@@ -109,10 +111,10 @@ def infer():
     _ = urlretrieve(video_url, video_path)
 
     frames, _, _ = read_video(str(video_path), output_format="TCHW")
-
+    print(f"FRAME BEFORE: {frames[100]}")
     img1_batch = torch.stack([frames[100]])
     img2_batch = torch.stack([frames[101]])
-
+    print(f"FRAME AFTER: {img1_batch}")
     weights = Raft_Large_Weights.DEFAULT
     transforms = weights.transforms()