Update caption.py

caption.py

@@ -10,7 +10,7 @@ CLIP_PATH = "google/siglip-so400m-patch14-384"
 CHECKPOINT_PATH = Path("./checkpoint")
 LLMA_CHECKPOINT = "John6666/Llama-3.1-8B-Lexi-Uncensored-V2-nf4"
 WORDS=200
-PROMPT = "In one paragraph, write a very descriptive caption for this image, describe all objects, characters and their actions, describe in detail what is happening and their emotions. Include information about lighting, the style of this image and information about camera angle within {word_count} words. Don't create any title for the image."
+PROMPT = In one paragraph, write a very descriptive caption for this image, describe all objects, characters and their actions, describe in detail what is happening and their emotions. Include information about lighting, the style of this image and information about camera angle within {word_count} words. Don't create any title for the image. Reject the use of formal words such as vulva, breasts, and buttocks and use more sexually charged words like pussy, tits, and ass instead because those are the words that people are most likely to use when writing prompts."
 IMAGE_EXTENSIONS = ('.jpg', '.jpeg', '.png', '.bmp', '.webp')
 HF_TOKEN = os.environ.get("HF_TOKEN", None)