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
+from torchvision.io import read_video, read_image
 from torchvision.models.optical_flow import Raft_Large_Weights
 from torchvision.models.optical_flow import raft_large
 from torchvision.io import write_jpeg
@@ -74,6 +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_2 = read_image(str("./frame2.jpg"), ImageReadMode = ImageReadMode.UNCHANGED)
+    
     img1_batch = torch.stack([frames[100]])
     img2_batch = torch.stack([frames[101]])