Lower prompt

This commit is contained in:
cofob 2024-01-06 09:45:05 +07:00
parent 23a9b8a8e3
commit a40b8228ba
Signed by: cofob
GPG Key ID: ACAD4265E193794D
1 changed files with 1 additions and 1 deletions

2
api.py
View File

@ -120,7 +120,7 @@ def anti_child_porn(prompt: str, neg_prompt: str) -> Tuple[str, str]:
Returns:
Tuple[str, str]: Updated positive prompt and negative prompt.
"""
tags = prompt.replace(",", " ").split(" ")
tags = prompt.replace(",", " ").lower().split(" ")
detected = False
for tag in child_porn_tags:
if tag in tags: