Update app.py

app.py

@@ -56,6 +56,7 @@ def generate(
     negative_prompt: str = "",
     use_negative_prompt: bool = False,
     seed: int = 0,
+    num_images_per_prompt: index = 1 ,
     width: int = 1024,
     height: int = 1024,
     guidance_scale: float = 3,
@@ -75,7 +76,7 @@ def generate(
         height=height,
         guidance_scale=guidance_scale,
         num_inference_steps=25,
-        num_images_per_prompt=1,
+        num_images_per_prompt=num_images_per_prompt,
         cross_attention_kwargs={"scale": 0.65},
         output_type="pil",
     ).images
@@ -109,13 +110,17 @@ def run():
     use_negative_prompt = data['use_negative_prompt']
     guidance_scale = data['guidance_scale']
     randomize_seed = data['randomize_seed']
+    num_images_per_prompt = data['num_images_per_prompt']
+    width = data['width']
+    height = data['height']
     result = generate(
     prompt,
     negative_prompt,
     use_negative_prompt,
     0,
-    1024,
-    1024,
+    num_images_per_prompt
+    width,
+    height,
     guidance_scale,
     randomize_seed)
     return jsonify({'out': result})