Update app.py

app.py

@@ -79,8 +79,8 @@ def infer():
     input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
     
-    img1_batch = torch.stack([frames[100]])
-    img2_batch = torch.stack([frames[101]])
+    img1_batch = torch.stack([input_frame_1])
+    img2_batch = torch.stack([input_frame_2])
     
     print(f"FRAME AFTER stack: {img1_batch}")
     
@@ -170,7 +170,7 @@ def infer():
     transform = T.ToPILImage()
     
     # convert the tensor to PIL image using above transform
-    img = transform(frames[101])
+    img = transform(input_frame_2)
     img = img.resize((960, 520))
     # display the PIL image
     #img.show()