Update app.py

app.py

@@ -27,14 +27,14 @@ def process_images(videos, x, y):
     images = images[:11]
     # transform images to a list of images tensor
     images = [torch.from_numpy(img).permute(2, 0, 1).float().to(device).unsqueeze(0) / 255.0 for img in images]
-    # if the max size of the image is larger than 1024, resize the image to 1024 with same ratio
+    # if the max size of the image is larger than 1024, resize the image to 768 with same ratio
     max_size = max(images[0].shape[2], images[0].shape[3])
     if max_size > 768:
         ratio = 768 / max_size
         images = [torch.nn.functional.interpolate(img, scale_factor=ratio, mode='bicubic', align_corners=True) for img
                   in images]
     # transform color image to gray image
-
+    
     result = model.forward_viz(images, layer=7, x=x, y=y)
     flow = result['flow']
     attention = result['attention']