Update app.py

app.py

@@ -17,10 +17,14 @@ for model_name in model_list:
 
 
 @spaces.GPU
-def generate_image(model_name, prompt, negative_prompt, num_inference_steps, guidance_scale):
+def generate_images(model_name, prompt, negative_prompt, num_inference_steps, guidance_scale, num_images=4):
     pipe = models[model_name]
-    output = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale)["images"][0]
-    return output
+    outputs = []
+    for _ in range(num_images):
+        output = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale)["images"][0]
+        outputs.append(output)
+    return outputs
+
 
     
 
@@ -35,12 +39,13 @@ with gr.Blocks() as demo:
                     negative_prompt = gr.Textbox(label="Negative Prompt", value="nsfw, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], kid, kid looking, child, childish look")
                     num_inference_steps = gr.Slider(minimum=10, maximum=50, step=1, value=25, label="Number of Inference Steps")
                     guidance_scale = gr.Slider(minimum=1, maximum=20, step=0.5, value=7.5, label="Guidance Scale")
+                    num_images = gr.Slider(minimum=1, maximum=4, step=1, value=4, label="Number of Images")
                 generate_btn = gr.Button("Generate Image")
                 
         with gr.Column():
             output_image = gr.Image(label="Generated Image", height=480, scale=1)
             
 
-    generate_btn.click(generate_image, inputs=[model_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale], outputs=output_image)
+    generate_btn.click(generate_image, inputs=[model_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale, num_images], outputs=output_image)
 
 demo.launch()