update app

app.py

@@ -30,7 +30,7 @@ prompt = f"A photo of {unique_id} {class_label} in a bucket"
 def infer(prompt, negative_prompt, guidance_scale=10, num_inference_steps=50):
     neg = negative_prompt if negative_prompt else None
     imgs = []
-    while len(imgs) < 2:
+    while len(imgs) < 3:
         next_prompt = pipeline(prompt, negative_prompt=neg, guidance_scale=guidance_scale, num_inference_steps=num_inference_steps, num_images_per_prompt=2)
         if not next_prompt.nsfw_content_detected[0]:
             imgs.append(next_prompt.images[0])