Update app.py

app.py

@@ -107,11 +107,15 @@ css = """
 theme = "Base"
 
 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("puppies in a cloud, 4k", placeholder="Prompt", show_label=False, lines=3)
-            negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=False, lines=3)
+            prompt = gr.Textbox(label="Prompt", placeholder="beautiful cat, 8k", show_label=False, 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=False, lines=3)
         with gr.Column():
             text_button = gr.Button("Generate", variant='primary', elem_id="generate")
     with gr.Row():
@@ -123,7 +127,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Tab("Generation"):
                 with gr.Row():
                     with gr.Column(scale=1):
-                        sampler = gr.Dropdown(value="Euler a", show_label=True, label="Sampling Method", choices=[
+                        sampler = gr.Dropdown(value="DPM++ SDE", show_label=True, label="Sampler", choices=[
                             "Euler",
                             "Euler a",
                             "LMS",
@@ -146,7 +150,7 @@ with gr.Blocks(css=css) as demo:
                         ])
                             
                     with gr.Column(scale=1):
-                        steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=50, value=25, step=1)
+                        steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=25, step=1)
 
                 with gr.Row():
                     with gr.Column(scale=1):
@@ -155,14 +159,15 @@ with gr.Blocks(css=css) as demo:
                         
                     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)
+                        batch_count = gr.Slider(label="Batch Count", maximum=4, minimum=1, value=2, step=1)
 
                 cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
-                seed = gr.Number(label="Seed", value=-1)
+                seed = gr.Number(label="Seed", value=-1, info = """'-1' is a random seed""")
 
                 
         with gr.Column(scale=2):
-            image_output = gr.Image()
+            image_output = gr.Gallery(label="Result Image(s)", show_download_button=False, show_share_button=True)
+
 
     text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)