ban nsfw

app.py

@@ -2,15 +2,17 @@ import gradio as gr
 import requests
 import json
 import time
-from profanity_check import predict  # For NSFW detection
+from better_profanity import profanity  
 
 url = "https://nexra.aryahcr.cc/api/image/complements"
 headers = {
     "Content-Type": "application/json"
 }
 
+profanity.load_censor_words()
+
 def is_nsfw(text):
-    return predict([text])[0] == 1  
+    return profanity.contains_profanity(text)
 
 def generate_image(prompt, model):
     if is_nsfw(prompt):