Update app.py

app.py

@@ -104,68 +104,74 @@ css = """
 }
 """
 
-theme = "Base"
+with gr.Blocks(css=css, theme="Base") as demo:
 
-with gr.Blocks(css=css) as demo:
- with gr.Row():
- gr.Markdown("""
- <h1><center>Stable Diffusion</center></h1>
 
- """)
- with gr.Row():
- with gr.Column(scale=6, min_width=600):
- prompt = gr.Textbox(label="Prompt", placeholder="beautiful cat, 8k", show_label=True, lines=2)
- negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", placeholder="What you don't want to see in the image", show_label=True, lines=3)
- with gr.Column():
- text_button = gr.Button("Generate", variant='primary', elem_id="generate")
- 
+ 
  with gr.Row():
- 
- 
+ gr.Markdown("<h1><center>Stable Diffusion Demo</center></h1>")
+ with gr.Tab("Playground"):
+ with gr.Row():
+ with gr.Column(scale=6, min_width=600):
+ prompt = gr.Textbox(label="Prompt", placeholder="beautiful cat, 8k", show_label=True, lines=2)
+ negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", placeholder="text, blurry, fuzziness", show_label=True, lines=3)
+ with gr.Column():
+ text_button = gr.Button("Generate", variant='primary', elem_id="generate")
+ 
+ with gr.Row():
+ with gr.Column(scale=3):
+
  
- with gr.Column(scale=2):
- image_output = gr.Image()
- 
- with gr.Accordion("Advanced options", open=False):
- with gr.Column(scale=6):
- model = gr.Dropdown(interactive=True,value="v1-5-pruned-emaonly.safetensors [d7049739]", show_label=True, label="Model", choices=prodia_client.list_models())
-
- with gr.Column(scale=1):
- sampler = gr.Dropdown(value="DPM++ SDE", show_label=True, label="Sampler", choices=[
- "Euler",
- "Euler a",
- "LMS",
- "Heun",
- "DPM2",
- "DPM2 a",
- "DPM++ 2S a",
- "DPM++ 2M",
- "DPM++ SDE",
- "DPM fast",
- "DPM adaptive",
- "LMS Karras",
- "DPM2 Karras",
- "DPM2 a Karras",
- "DPM++ 2S a Karras",
- "DPM++ 2M Karras",
- "DPM++ SDE Karras",
- "DDIM",
- "PLMS",
- ])
+ with gr.Column(scale=2):
+ image_output = gr.Image()
+
+ with gr.Accordion("Advanced options", open=False):
+ with gr.Row():
+ with gr.Column(scale=6):
+ model = gr.Dropdown(interactive=True,value="v1-5-pruned-emaonly.safetensors [d7049739]", show_label=True, label="Stable Diffusion Checkpoint", choices=prodia_client.list_models())
+
+ 
+ with gr.Row():
+ with gr.Column(scale=1):
+ sampler = gr.Dropdown(value="DPM++ SDE", show_label=True, label="Sampler", choices=[
+ "Euler",
+ "Euler a",
+ "LMS",
+ "Heun",
+ "DPM2",
+ "DPM2 a",
+ "DPM++ 2S a",
+ "DPM++ 2M",
+ "DPM++ SDE",
+ "DPM fast",
+ "DPM adaptive",
+ "LMS Karras",
+ "DPM2 Karras",
+ "DPM2 a Karras",
+ "DPM++ 2S a Karras",
+ "DPM++ 2M Karras",
+ "DPM++ SDE Karras",
+ "DDIM",
+ "PLMS",
+ ])
  
- with gr.Column(scale=1):
- steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=30, step=1, interactive=True)
+ with gr.Column(scale=1):
+ steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=30, step=1)
 
  with gr.Row():
  with gr.Column(scale=1):
  width = gr.Slider(label="Width", maximum=1024, value=512, step=8)
  height = gr.Slider(label="Height", maximum=1024, value=512, step=8)
  
- cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
- seed = gr.Number(label="Seed", value=-1, info = """'-1' is a random seed""")
+ with gr.Column(scale=1):
+ batch_size = gr.Slider(label="Batch Size", maximum=1, value=1)
+ batch_count = gr.Slider(label="Batch Count", maximum=1, value=1)
 
+ cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
+ seed = gr.Number(label="Seed", value=-1, info="""'-1' is random seed""")
 
- text_button.click(flip_text, inputs=[prompt, negative_prompt, model, sampler, cfg_scale, width, height, seed], outputs=image_output)
+ 
+ text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
  
 demo.queue(concurrency_count=10)
 demo.launch()