Update app.py

app.py

@@ -108,7 +108,8 @@ def query(lora_id, prompt, is_negative=False, steps=28, cfg_scale=3.5, sampler="
     except Exception as e:
         print(f"Error when trying to open the image: {e}")
         return None
-
+    NUM_IMAGES_PER_PROMPT = 1
+    
 examples = [
     "a beautiful woman with blonde hair and blue eyes",
     "a beautiful woman with brown hair and grey eyes",
@@ -168,6 +169,8 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                             method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "DPM Fast" "Euler", "Euler a", "Euler+beta", "Heun", "DDIM", "PLMS", "UniPC"])
                             strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
                             seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
+
+                            num_images_per_prompt = gr.Slider(label="Images", minimum=1, maximum=5, step=1, value=2, visible=True)
                     with gr.Row():
                         with gr.Accordion("🫘Seed", open=False):
                             seed_output = gr.Textbox(label="Seed Used", show_copy_button = True, elem_id="seed-output")