Update app.py

app.py

@@ -147,7 +147,7 @@ def infer():
     #img1_batch = torch.stack([frames[0]])
     #img2_batch = torch.stack([frames[1]])
 
-    img1_batch = torch.stack([input_frame_1])
+    img1_batch = torch.stack([input_diffused])
     img2_batch = torch.stack([input_frame_2])
     
     print(f"FRAME AFTER stack: {img1_batch}")