Rooni commited on
Commit
7304133
·
1 Parent(s): ea791ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -28,14 +28,18 @@ def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
28
 
29
  return generated_image
30
 
 
 
 
31
  iface = gr.Interface(
32
- fn=generate_image,
33
  inputs=[
34
  gr.Textbox("text", label="Prompt"),
35
  gr.Textbox("text", label="Negative Prompt"),
36
  gr.Slider(minimum=1, maximum=30, label="Sampling Steps"),
37
  gr.Slider(minimum=0.1, maximum=2.0, label="CFG Scale"),
38
- gr.Textbox("number", label="Seed (default -1)")
 
39
  ],
40
  outputs=gr.Image(label="Generated Image"),
41
  live=True,
 
28
 
29
  return generated_image
30
 
31
+ def generate_on_button_click(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
32
+ return generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed)
33
+
34
  iface = gr.Interface(
35
+ fn=generate_on_button_click,
36
  inputs=[
37
  gr.Textbox("text", label="Prompt"),
38
  gr.Textbox("text", label="Negative Prompt"),
39
  gr.Slider(minimum=1, maximum=30, label="Sampling Steps"),
40
  gr.Slider(minimum=0.1, maximum=2.0, label="CFG Scale"),
41
+ gr.Textbox("number", label="Seed (default -1)"),
42
+ gr.Button("Generate", generate_on_button_click)
43
  ],
44
  outputs=gr.Image(label="Generated Image"),
45
  live=True,