Update app.py

app.py

@@ -57,6 +57,7 @@ def generate(
     prompt: str,
     negative_prompt: str = "",
     use_negative_prompt: bool = False,
+    num_inference_steps: int = 30,
     seed: int = 0,
     width: int = 1024,
     height: int = 1024,
@@ -77,7 +78,7 @@ def generate(
         width=width,
         height=height,
         guidance_scale=guidance_scale,
-        num_inference_steps=25,
+        num_inference_steps=num_inference_steps,
         num_images_per_prompt=1,
         cross_attention_kwargs={"scale": 0.65},
         output_type="pil",
@@ -131,6 +132,14 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
             placeholder="Enter a negative prompt",
             visible=True,
         )
+        with gr.Row():
+            guidance_scale = gr.Slider(
+                label="steps",
+                minimum=8,
+                maximum=35,
+                step=0.1,
+                value=25,
+            )
         seed = gr.Slider(
             label="Seed",
             minimum=0,
@@ -154,7 +163,7 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
                 maximum=2048,
                 step=8,
                 value=1024,
-            )
+            )num_inference_steps
         with gr.Row():
             guidance_scale = gr.Slider(
                 label="Guidance Scale",
@@ -164,6 +173,8 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
                 value=6,
             )
 
+        
+
     gr.Examples(
         examples=examples,
         inputs=prompt,
@@ -191,6 +202,7 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
             prompt,
             negative_prompt,
             use_negative_prompt,
+            num_inference_steps,
             seed,
             width,
             height,