Krebzonide commited on
Commit
be13023
·
1 Parent(s): ff8efdc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,7 +39,7 @@ def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tq
39
  guidance_scale=0.0,
40
  generator=torch.manual_seed(seed),
41
  ).images
42
- return gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(images)], height=height+90), seed
43
 
44
  def set_base_model():
45
  pipe = AutoPipelineForText2Image.from_pretrained(
@@ -56,7 +56,7 @@ with gr.Blocks(css=css) as demo:
56
  prompt = gr.Textbox(label="Prompt")
57
  submit_btn = gr.Button("Generate", elem_classes="btn-green")
58
  with gr.Row():
59
- samp_steps = gr.Slider(1, 30, value=10, step=1, label="Sampling steps")
60
  batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size", interactive=True)
61
  seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
62
  lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
 
39
  guidance_scale=0.0,
40
  generator=torch.manual_seed(seed),
41
  ).images
42
+ return gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(images)]), seed
43
 
44
  def set_base_model():
45
  pipe = AutoPipelineForText2Image.from_pretrained(
 
56
  prompt = gr.Textbox(label="Prompt")
57
  submit_btn = gr.Button("Generate", elem_classes="btn-green")
58
  with gr.Row():
59
+ samp_steps = gr.Slider(1, 20, value=5, step=1, label="Sampling steps")
60
  batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size", interactive=True)
61
  seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
62
  lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)