Update app.py

app.py

@@ -73,8 +73,8 @@ def get_canny_filter(image):
     return canny_image
 
 
-def generate_images(image, prompt):
-    canny_image = get_canny_filter(image)
+def generate_images(prompt, canny_image):
+    
     output = pipe(
         prompt,
         canny_image,
@@ -125,7 +125,8 @@ def infer():
     #print(f"FRAME BEFORE stack: {frames[100]}")
     prompt = "astronaut crew inside a spaceship, artwork by Claude Monet, beautiful details"
     pil2diff_img = Image.open("./frame1.jpg")
-    diffused_img = generate_images(pil2diff_img, prompt)
+    canny_image = get_canny_filter(pil2diff_img)
+    diffused_img = generate_images(prompt, canny_image)
     print(f"DIFFUSED IMG: {diffused_img[1]}")
     
     diffused_img[1].save("diffused_input1.jpg")
@@ -266,7 +267,8 @@ def infer():
     blend2.save("blended2.jpg")
 
     pil2diff_img = Image.open("blended2.jpg")
-    diffused_img = generate_images(pil2diff_img, prompt)
+    canny_image = get_canny_filter(pil2diff_img)
+    diffused_img = generate_images(prompt, canny_image)
     print(f"DIFFUSED IMG: {diffused_img[1]}")
     
     diffused_img[1].save("diffused_blended_2.jpg")