Update app.py

app.py

@@ -36,6 +36,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
 
 pipe.to(device)
 
+
 # 📸 Edge detection function using OpenCV (Canny)
 @spaces.GPU
 def apply_canny(image, low_threshold, high_threshold):
@@ -45,6 +46,7 @@ def apply_canny(image, low_threshold, high_threshold):
     image = np.concatenate([image, image, image], axis=2)
     return Image.fromarray(image)
 
+
 # 🎨 Image generation function from image
 @spaces.GPU
 def generate_image(prompt, input_image, low_threshold, high_threshold, strength, guidance, controlnet_conditioning_scale):
@@ -64,6 +66,7 @@ def generate_image(prompt, input_image, low_threshold, high_threshold, strength,
     
     return edge_detected, result
 
+
 # 🎨 Image generation function from prompt
 @spaces.GPU
 def generate_prompt(prompt, strength, guidance):
@@ -111,8 +114,8 @@ with gr.Blocks() as demo:
     )
 
     generate_prompt_button.click(
-        fn=generate_image,
-        inputs=[prompt, input_image, low_threshold, high_threshold, strength, guidance, controlnet_conditioning_scale],
+        fn=generate_prompt,
+        inputs=[prompt, strength, guidance],
         outputs=[result_output]
     )