Update app.py

app.py

@@ -138,16 +138,16 @@ def infer():
 
     canny_image.save("canny1.jpg")
     canny_image2.save("canny2.jpg")
-    input_frame_1 = read_image(str("diffused_input1.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_1 = read_image(str("./basket1.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")
     input_frame_2 = read_image(str("./basket2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
-    
+    input_diffused = read_image(str("diffused_input1.jpg"), ImageReadMode.UNCHANGED)
     #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_2])
+    img2_batch = torch.stack([input_diffused])
     
     print(f"FRAME AFTER stack: {img1_batch}")