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", num_images_per_prompt =1, 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, 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
     # 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)
@@ -86,6 +86,7 @@ def query(lora_id, prompt, is_negative=False, steps=28,  cfg_scale=3.5, sampler=
         "cfg_scale": cfg_scale,
         "seed": seed if seed != -1 else random.randint(1, 1000000000),
         "strength": strength,
+        "num_images_per_prompt": NUM_IMAGES_PER_PROMPT,
         "parameters": {
             "width": width,  # Pass the width to the API
             "height": height  # Pass the height to the API