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", num_images_per_prompt =1, seed=-1, strength=0.7, width=1024, height=1024):
     if prompt == "" or prompt == None:
         return None
 
@@ -108,7 +108,7 @@ 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",