Update app.py

app.py

@@ -82,8 +82,8 @@ def infer():
     #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)
+    img1_batch = torch.stack([input_frame_1])
+    img2_batch = torch.stack([input_frame_2])
     
     print(f"FRAME AFTER stack: {img1_batch}")