test gradio

app.py

@@ -31,8 +31,7 @@ def generate_image(prompt, reference_image, controlnet_conditioning_scale):
         control_image=reference_image.resize((512, 512)),
         controlnet_conditioning_scale=controlnet_conditioning_scale,
         guidance_scale=7.5,
-        num_inference_steps=75  # Increased from 50 to refine quality
-    ).images[0]
+    )
     return generated_image
 
 # Set up Gradio interface
@@ -44,8 +43,8 @@ interface = gr.Interface(
         gr.Slider(label="Control Net Conditioning Scale", minimum=0, maximum=1.0, step=0.1, value=0.6),
     ],
     outputs="image",
-    title="Image Generation with Stable Diffusion 3.5 and ControlNet",
-    description="Generates an image based on a text prompt and a reference image using Stable Diffusion 3.5 with ControlNet."
+    title="Image Generation with Stable Diffusion 3 medium and ControlNet",
+    description="Generates an image based on a text prompt and a reference image using Stable Diffusion 3 medium with ControlNet."
 )
 
 # Launch the Gradio interface