more nsfw words

Signed-off-by: Jirka Kremser <jiri.kremser@gmail.com>
kedify · Jun 20, 2024 · 6067571 · 6067571
1 parent 5f7e0a6
commit 6067571
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/samples/stable-diffusion/webui/app/prompt.tsx b/samples/stable-diffusion/webui/app/prompt.tsx
@@ -31,7 +31,7 @@ export const Prompt: React.FC<PromptProps> = ({ sendToQ }) => {
     },
   });
   const safetyCheck = (prompt: string) => {
-    const stopList = ["boob", "tits", "vagina", "penis", "dick", "naked", "nude", "fuck", "coitus", "sex", "shit", "blood", "murder", "dead"];
+    const stopList = ["boob", "tits", "vagina", "penis", "dick", "cock", "trump", "naked", "nude", "fuck", "coitus", "sex", "shit", "blood", "murder", "dead"];
     return stopList.some(w => prompt.toLowerCase().includes(w));
   }