Update app.py

app.py

@@ -74,13 +74,16 @@ def infer():
     frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
     print(f"FRAME BEFORE stack: {frames[100]}")
     
-    #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.UNCHANGED)
-    #print(f"FRAME 1: {input_frame_2}")
+    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.UNCHANGED)
+    print(f"FRAME 1: {input_frame_2}")
     
-    img1_batch = torch.stack([frames[0]])
-    img2_batch = torch.stack([frames[1]])
+    #img1_batch = torch.stack([frames[0]])
+    #img2_batch = torch.stack([frames[1]])
+
+    img1_batch = torch.stack(input_frame_1])
+    img2_batch = torch.stack(input_frame_2])
     
     print(f"FRAME AFTER stack: {img1_batch}")
     
@@ -170,7 +173,8 @@ def infer():
     transform = T.ToPILImage()
     
     # convert the tensor to PIL image using above transform
-    img = transform(frames[1])
+    #img = transform(frames[1])
+    img = transform(input_frame_2)
     img = img.resize((960, 520))
     # display the PIL image
     #img.show()