Update app.py

app.py

@@ -242,7 +242,12 @@ def spirit_animal_baseline(image_path, num_images = 4):
             for _ in range(num_images):
                 images = pipe_controlnet(
                     prompt=prompt,
-                    negative_prompt="multiple heads, extra limbs, duplicate faces, mutated anatomy, disfigured, blurry",
+                    negative_prompt=(
+                    "multiple heads, two heads, double head, triple head, extra limbs, extra arms, extra legs, "
+                    "duplicate faces, multiple faces, mutated anatomy, deformed, disfigured, malformed, "
+                    "extra ears, fused ears, blurred, low resolution, cartoonish, watermark, text, logo, "
+                    "poorly drawn, distorted, floating limbs, out-of-frame"
+                    ),
                     num_inference_steps=20,
                     image=pose_pil,
                     guidance_scale=5,
@@ -329,7 +334,12 @@ def spirit_animal_with_background(image_path, num_images = 4):
             for _ in range(num_images):
                 images = pipe_inpaint_controlnet(
                     prompt=prompt,
-                    negative_prompt="multiple heads, extra limbs, duplicate faces, mutated anatomy, disfigured, blurry",
+                    negative_prompt=(
+                        "multiple heads, two heads, double head, triple head, extra limbs, extra arms, extra legs, "
+                        "duplicate faces, multiple faces, mutated anatomy, deformed, disfigured, malformed, "
+                        "extra ears, fused ears, blurred, low resolution, cartoonish, watermark, text, logo, "
+                        "poorly drawn, distorted, floating limbs, out-of-frame"
+                    ),
                     num_inference_steps=20,
                     image=Image.fromarray(image_rgb),
                     mask_image=expanded_mask,
@@ -392,10 +402,12 @@ def generate_multiple_animals(image_path, keep_background=True, num_images = 4):
     if 'choices' in response_json and len(response_json['choices']) > 0:
       content = response_json['choices'][0]['message']['content']
       prompts = [prompt.strip() for prompt in content.strip().split('.') if prompt.strip()]
-      negative_prompt = (
-          "multiple heads, extra limbs, duplicate faces, mutated anatomy, disfigured, "
-          "blurry, deformed, text, watermark, logo, low resolution"
-      )
+      negative_prompt=(
+            "multiple heads, two heads, double head, triple head, extra limbs, extra arms, extra legs, "
+            "duplicate faces, multiple faces, mutated anatomy, deformed, disfigured, malformed, "
+            "extra ears, fused ears, blurred, low resolution, cartoonish, watermark, text, logo, "
+            "poorly drawn, distorted, floating limbs, out-of-frame")
+
       formatted_prompts = "\n".join(f"{i+1}. {prompt}" for i, prompt in enumerate(prompts))
 
     with mp_pose.Pose(static_image_mode=True) as pose: