Update app.py

app.py

@@ -125,9 +125,9 @@ def infer():
     #frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
     #print(f"FRAME BEFORE stack: {frames[100]}")
     
-    prompt = "wltzwthbshr astronaut crew inside a spaceship"
+    prompt = "wltzwthbshr basketball player"
     
-    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, pil2diff_img)
     print(f"DIFFUSED IMG: {diffused_img[1]}")
@@ -139,9 +139,9 @@ def infer():
 
     canny_image.save("canny1.jpg")
     canny_image2.save("canny2.jpg")
-    input_frame_1 = read_image(str("./frame1.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("./frame2.jpg"), ImageReadMode.UNCHANGED)
+    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]])