Spaces:
Running
on
T4
Running
on
T4
Commit
·
be13023
1
Parent(s):
ff8efdc
Update app.py
Browse files
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)]
|
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,
|
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)
|