Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image
 from generationPipeline import generate
 from transformers import CLIPTokenizer
 from loadModel import preload_models_from_standard_weights
-from profanity_check import predict  # Import the profanity-check library
+from better_profanity import profanity  # Import the profanity-check library
 import gradio as gr
 
 
@@ -18,9 +18,11 @@ model_file = "weights-inkpen.ckpt"
 models = preload_models_from_standard_weights(model_file, Device)
 
 ## profanity check on input prompt
+profanity.load_censor_words() 
+
 
 def filter_prompt(prompt):
-    if predict([prompt])[0] == 1:
+    if profanity.contains_profanity(prompt):
         return "Inappropriate content detected. Please modify the input."
     return prompt