test gradio

app.py

@@ -31,22 +31,20 @@ 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,
-    )
+    ).images[0]
     return generated_image
 
-# # Set up Gradio interface
-# interface = gr.Interface(
-#     fn=generate_image,
-#     inputs=[
-#         gr.Textbox(label="Prompt"),
-#         gr.Image( type= "pil",label="Reference Image (Style)"),
-#         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 medium and ControlNet",
-#     description="Generates an image based on a text prompt and a reference image using Stable Diffusion 3 medium with ControlNet."
-# )
+# Set up Gradio interface
+interface = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.Textbox(label="Prompt"),
+        gr.Image( type= "pil",label="Reference Image (Style)"),
+        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 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
-# interface.launch()
-gr.Interface.from_pipeline(pipe).launch()
+interface.launch()