Update app.py

app.py

@@ -52,7 +52,7 @@ headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 
     
-def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler="DPM++ 2M Karras", seed=-1, num_images_per_prompt =1, strength=0.7, width=1024, height=1024):
     if prompt == "" or prompt == None:
         return None
 
@@ -65,7 +65,7 @@ def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler=
 
     API_TOKEN = random.choice([os.getenv("HF_READ_TOKEN")])
     headers = {"Authorization": f"Bearer {API_TOKEN}"}
-    NUM_IMAGES_PER_PROMPT=1
+    #NUM_IMAGES_PER_PROMPT=1
     # prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
     # print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')
     prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
@@ -190,7 +190,7 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                 inputs = [text_prompt],
             )
             
-    text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, num_images_per_prompt, method, seed, strength, width, height], outputs=[image_output, seed_output])
+    text_button.click(query, inputs=[custom_lora, text_prompt, negative_prompt, steps, cfg, num_images_per_prompt, method, seed, num_images_per_prompt, strength, width, height], outputs=[image_output, seed_output])
 
     with gr.Tab("Image Upscaler"):    
         with gr.Row():