Update app.py

app.py

@@ -62,6 +62,7 @@ def generate(
     width: int = 1024,
     height: int = 1024,
     guidance_scale: float = 3,
+    inference_steps: int = 50,
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ):
@@ -118,21 +119,27 @@ with gr.Blocks(css=css) as demo:
             prompt = gr.Text(
                 label="Prompt",
                 show_label=False,
-                max_lines=1,
                 placeholder="Enter your prompt",
                 container=False,
             )
             run_button = gr.Button("Run", scale=0)
         result = gr.Gallery(label="Result", columns=1, preview=True, show_label=False)
-    with gr.Accordion("Advanced options", open=False):
+    with gr.Accordion("Advanced options", open=True):
         with gr.Row():
             use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=False)
             negative_prompt = gr.Text(
                 label="Negative prompt",
-                max_lines=1,
                 placeholder="Enter a negative prompt",
                 visible=False,
             )
+        with gr.Row():
+            inference_steps = gr.Slider(
+                label="Sampling Steps",
+                minimum=1,
+                maximum=60,
+                step=1,
+                value=50
+            )
         seed = gr.Slider(
             label="Seed",
             minimum=0,
@@ -146,16 +153,16 @@ with gr.Blocks(css=css) as demo:
             width = gr.Slider(
                 label="Width",
                 minimum=512,
-                maximum=1048,
+                maximum=1536,
                 step=8,
-                value=768,
+                value=1024,
             )
             height = gr.Slider(
                 label="Height",
                 minimum=512,
-                maximum=1048,
+                maximum=1536,
                 step=8,
-                value=768,
+                value=1024,
             )
         with gr.Row():
             guidance_scale = gr.Slider(
@@ -197,6 +204,7 @@ with gr.Blocks(css=css) as demo:
             width,
             height,
             guidance_scale,
+            inference_steps,
             randomize_seed,
         ],
         outputs=[result, seed],