Update app.py

app.py

@@ -87,32 +87,20 @@ def generate_image(prompt, input_image, low_threshold, high_threshold, strength,
 
     print(pipe.get_active_adapters())
 
-    ## execute if an image is provided
-    if not isinstance(input_image, type(None)):
-        # Apply edge detection
-        edge_detected = apply_canny(input_image, low_threshold, high_threshold)
-        
-        # Generate styled image using ControlNet
-        result = pipe(
-            prompt=prompt,
-            image=edge_detected,
-            num_inference_steps=30,
-            guidance_scale=guidance,
-            controlnet_conditioning_scale=float(controlnet_conditioning_scale),
-            strength=strength
-        ).images[0]
-        
-        return edge_detected, result
-
-    else:
-        # Generate styled image using ControlNet
-        result = pipe(
-            prompt=prompt,
-            num_inference_steps=30,
-            guidance_scale=guidance
-        ).images[0]
-
-        return None, result
+    # Apply edge detection
+    edge_detected = apply_canny(input_image, low_threshold, high_threshold)
+    
+    # Generate styled image using ControlNet
+    result = pipe(
+        prompt=prompt,
+        image=edge_detected,
+        num_inference_steps=30,
+        guidance_scale=guidance,
+        controlnet_conditioning_scale=float(controlnet_conditioning_scale),
+        strength=strength
+    ).images[0]
+    
+    return edge_detected, result
 
 # 🖥️ Gradio UI
 with gr.Blocks() as demo: