.

image_moderator.py

@@ -68,7 +68,8 @@ from transformers import pipeline
 from better_profanity import profanity
 
 # Load profanity model
-pipeline("text-classification", model="unitary/toxic-bert").save_pretrained("./models/unitary-toxic-bert")
+# pipeline("text-classification", model="unitary/toxic-bert").save_pretrained("./models/unitary-toxic-bert")
+toxic_classifier = pipeline("text-classification", model="./models/unitary-toxic-bert")
 profanity.load_censor_words()
 custom_words = [
     "idiot", "moron", "dumb", "stupid", "loser", "bastard", "retard", "scumbag",