diff --git a/api.py b/api.py index f7d36e6..d105c96 100644 --- a/api.py +++ b/api.py @@ -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: