Update app.py

app.py

@@ -126,7 +126,7 @@ def infer():
     
     prompt = "astronaut crew inside a spaceship, artwork by Claude Monet, beautiful details"
     
-    pil2diff_img = Image.open("./frame1.jpg")
+    pil2diff_img = Image.open("./basket1.jpg")
     canny_image = get_canny_filter(pil2diff_img)
     diffused_img = generate_images(prompt, canny_image)
     print(f"DIFFUSED IMG: {diffused_img[1]}")
@@ -140,7 +140,7 @@ def infer():
     canny_image2.save("canny2.jpg")
     input_frame_1 = read_image(str("diffused_input1.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")
-    input_frame_2 = read_image(str("./frame2.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_2 = read_image(str("./basket2.jpg"), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
     
     #img1_batch = torch.stack([frames[0]])