Update app.py

app.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python
 import gradio as gr
 import requests
 import io
@@ -79,6 +80,8 @@ API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 
+NUM_IMAGES_PER_PROMPT = 1
+
 def flip_image(x):
     return np.fliplr(x)
 
@@ -122,6 +125,7 @@ def query(lora_id, prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="
         "parameters": {
             "width": width,  # Pass the width to the API
             "height": height  # Pass the height to the API
+        "num_images_per_prompt": NUM_IMAGES_PER_PROMPT,    
         }
     }
 
@@ -200,6 +204,7 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                             cfg = gr.Slider(label="CFG Scale", value=3.5, minimum=1, maximum=20, step=0.5)
                             method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM2 Ancestral", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "dpmpp_2s_ancestral", "Euler", "Euler CFG PP", "Euler a", "Euler Ancestral", "Euler+beta", "Heun", "Heun PP2", "DDIM", "PLMS", "UniPC", "UniPC BH2"])
                             strength = gr.Slider(label="Prompt Strength", value=100, minimum=0, maximum=100, step=1)
+                            num_images_per_prompt = gr.Slider(label="Images", minimum=1, maximum=5, step=1, value=2, visible=True)
                             seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
                     with gr.Row():
                         with gr.Accordion("🫘Seed", open=False):