bug fix[+]

app.py

@@ -58,6 +58,7 @@ def generate(
     negative_prompt: str = "",
     use_negative_prompt: bool = False,
     num_inference_steps: int = 30,
+    num_images_per_prompt: int = 2
     seed: int = 0,
     width: int = 1024,
     height: int = 1024,
@@ -79,7 +80,7 @@ def generate(
         height=height,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
-        num_images_per_prompt=1,
+        num_images_per_prompt=num_images_per_prompt,
         cross_attention_kwargs={"scale": 0.65},
         output_type="pil",
     ).images
@@ -135,11 +136,19 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
         with gr.Row():
             num_inference_steps = gr.Slider(
                 label="steps",
-                minimum=8,
+                minimum=10,
                 maximum=35,
-                step=0.1,
+                step=8,
                 value=25,
             )
+        with gr.Row():
+            num_images_per_prompt = gr.Slider(
+                label="images",
+                minimum=1,
+                maximum=5,
+                step=8,
+                value=2,
+            )
         seed = gr.Slider(
             label="Seed",
             minimum=0,
@@ -203,6 +212,7 @@ with gr.Blocks(css=css, theme="pseudolab/huggingface-korea-theme") as demo:
             negative_prompt,
             use_negative_prompt,
             num_inference_steps,
+            num_images_per_prompt,
             seed,
             width,
             height,