amos1088 commited on
Commit
e129330
·
1 Parent(s): 29ea1f8
Files changed (1) hide show
  1. app.py +32 -37
app.py CHANGED
@@ -66,40 +66,35 @@ def gui_generation(prompt, ref_img, guidance_scale, ipadapter_scale):
66
  ipadapter_scale=ipadapter_scale).images
67
  return image[0]
68
 
69
- # Create Gradio interface
70
- with gr.Blocks() as demo:
71
- gr.Markdown("# Stable Diffusion 3.5 Image Generation")
72
-
73
- with gr.Row():
74
- prompt_box = gr.Textbox(label="Prompt", placeholder="Enter your image generation prompt")
75
-
76
- with gr.Row():
77
- ref_img = gr.File(label="Upload Reference Image")
78
-
79
- with gr.Row():
80
- guidance_slider = gr.Slider(
81
- label="Guidance Scale",
82
- minimum=2,
83
- maximum=16,
84
- value=7,
85
- step=0.5,
86
- info="Controls adherence to the text prompt"
87
- )
88
- ipadapter_slider = gr.Slider(
89
- label="IP-Adapter Scale",
90
- minimum=0,
91
- maximum=1,
92
- value=0.5,
93
- step=0.1,
94
- info="Controls influence of the image prompt"
95
- )
96
-
97
- generate_btn = gr.Button("Generate")
98
-
99
- generate_btn.click(
100
- fn=gui_generation,
101
- inputs=[prompt_box, ref_img, guidance_slider, ipadapter_slider],
102
- outputs="image",
103
- )
104
-
105
- demo.launch()
 
66
  ipadapter_scale=ipadapter_scale).images
67
  return image[0]
68
 
69
+ prompt_box = gr.Textbox(label="Prompt", placeholder="Enter your image generation prompt")
70
+ ref_img = gr.File(label="Upload Reference Image")
71
+ guidance_slider = gr.Slider(
72
+ label="Guidance Scale",
73
+ minimum=2,
74
+ maximum=16,
75
+ value=7,
76
+ step=0.5,
77
+ info="Controls adherence to the text prompt")
78
+
79
+ ipadapter_slider = gr.Slider(
80
+ label="IP-Adapter Scale",
81
+ minimum=0,
82
+ maximum=1,
83
+ value=0.5,
84
+ step=0.1,
85
+ info="Controls influence of the image prompt"
86
+ )
87
+
88
+
89
+
90
+ # Set up Gradio interface
91
+ interface = gr.Interface(
92
+ fn=gui_generation,
93
+ inputs=[prompt_box, ref_img, guidance_slider, ipadapter_slider],
94
+ outputs="image",
95
+ title="Image Generation with Stable Diffusion 3 medium and ControlNet",
96
+ description="Generates an image based on a text prompt and a reference image using Stable Diffusion 3 medium with ControlNet."
97
+
98
+ )
99
+
100
+ interface.launch()