uuu

app.py

@@ -64,10 +64,7 @@ pipe.init_ipadapter(
 )
 
 @spaces.GPU
-def gui_generation(prompt, ref_img):
-    """
-    Generate images using Stable Diffusion 3.5
-    """
+def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
     # Ensure reference image is in the correct format
     if ref_img:
         ref_img = ref_img.convert("RGB")
@@ -85,15 +82,14 @@ def gui_generation(prompt, ref_img):
             prompt=prompt,
             negative_prompt="lowres, low quality, worst quality",
             num_inference_steps=24,
-            guidance_scale=5.0,
+            guidance_scale=guidance_scale,
             generator=torch.Generator("cuda").manual_seed(42),
             clip_image=ref_img_tensor,
-            ipadapter_scale=0.5,
+            ipadapter_scale=ipadapter_scale,
         ).images[0]
 
     return image
 
-
 # Create Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Stable Diffusion 3.5 Image Generation")
@@ -103,13 +99,32 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         ref_img = gr.Image(type="pil", label="Upload Reference Image")
-    gallery = gr.Image(type="pil", label="Generated Image")
+
+    with gr.Row():
+        guidance_slider = gr.Slider(
+            label="Guidance Scale",
+            minimum=2,
+            maximum=16,
+            value=7,
+            step=0.5,
+            info="Controls adherence to the text prompt"
+        )
+        ipadapter_slider = gr.Slider(
+            label="IP-Adapter Scale",
+            minimum=0,
+            maximum=1,
+            value=0.5,
+            step=0.1,
+            info="Controls influence of the image prompt"
+        )
 
     generate_btn = gr.Button("Generate")
+    gallery = gr.Image(type="pil", label="Generated Image")
 
     generate_btn.click(
         fn=gui_generation,
-        inputs=[prompt_box, ref_img],
+        inputs=[prompt_box, ref_img, guidance_slider, ipadapter_slider],
         outputs=gallery
     )
+
 demo.launch()