Merge pull request #32 from jkremser/safe_checker_logic

stable-diff: safety checker (negating the condition)
kedify · Jun 19, 2024 · 73a2931 · 73a2931
2 parents decc812 + 5f7e0a6
commit 73a2931
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/samples/stable-diffusion/webui/app/prompt.tsx b/samples/stable-diffusion/webui/app/prompt.tsx
@@ -31,13 +31,13 @@ export const Prompt: React.FC<PromptProps> = ({ sendToQ }) => {
     },
   });
   const safetyCheck = (prompt: string) => {
-    const stopList = ["boobs", "tits", "vagina", "penis", "dick", "naked", "nude", "fuck", "coitus", "sex", "shit", "blood", "murder", "dead"];
+    const stopList = ["boob", "tits", "vagina", "penis", "dick", "naked", "nude", "fuck", "coitus", "sex", "shit", "blood", "murder", "dead"];
     return stopList.some(w => prompt.toLowerCase().includes(w));
   }
 
   const generateOnClick = () => {
     if (prompt) {
-      if (!safetyCheck(prompt)) {
+      if (safetyCheck(prompt)) {
         toast.warning("Please don't use this tool for creating NSFW content.");
         focusPrompt();
         return