Update app.py

app.py

@@ -25,7 +25,7 @@ import torch
 from PIL import Image
 import matplotlib.pyplot as plt
 import torchvision.transforms.functional as F
-from torchvision.io import read_video, read_image
+from torchvision.io import read_video, read_image, ImageReadMode
 from torchvision.models.optical_flow import Raft_Large_Weights
 from torchvision.models.optical_flow import raft_large
 from torchvision.io import write_jpeg
@@ -74,9 +74,9 @@ def infer():
     frames, _, _ = read_video(str(video_path), output_format="TCHW")
     print(f"FRAME BEFORE stack: {frames[100]}")
     
-    input_frame_1 = read_image(str("./frame1.jpg"), ImageReadMode = ImageReadMode.UNCHANGED)
+    input_frame_1 = read_image(str("./frame1.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")
-    input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode = ImageReadMode.UNCHANGED)
+    input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
     
     img1_batch = torch.stack([frames[100]])