Update app.py

app.py

@@ -24,20 +24,17 @@ def preprocess_image(image):
     depth_map = np.clip(image, 0, 255)
     return Image.fromarray(depth_map)
 
-def generate_image(prompt, input_image):
+def generate_image(input_image):
     """Generate an image using Stable Diffusion ControlNet with IP-Adapter."""
     pipe = load_controlnet_model()
     processed_image = preprocess_image(input_image)
-    result = pipe(prompt, image=processed_image, num_inference_steps=50).images[0]
+    result = pipe(image=processed_image, num_inference_steps=30).images[0]
     return result
 
 # Gradio Interface
 demo = gr.Interface(
     fn=generate_image,
-    inputs=[
-        gr.Textbox(label="Enter your prompt"),
-        gr.Image(type="pil", label="Upload reference image")
-    ],
+    inputs=gr.Image(type="pil", label="Upload reference image"),
     outputs=gr.Image(type="pil", label="Generated Image"),
     title="Stable Diffusion with ControlNet and IP-Adapter",
     description="Generate images with precise object placement and consistent style using ControlNet and IP-Adapter.",