cachedir

image_moderator.py

@@ -4,7 +4,7 @@ from transformers import CLIPProcessor, CLIPModel, ViTForImageClassification, Vi
 from PIL import Image, ImageFilter
 import torch
 import cv2
-
+import os
 class ImageContentModerator:
     def __init__(self, nsfw_threshold=0.85, blur_radius=99):
         # NSFW Setup
@@ -66,9 +66,9 @@ class ImageContentModerator:
         return output_path
 from transformers import pipeline
 from better_profanity import profanity
-
+os.makedirs("./cache_dir", exist_ok=True)
 # Load profanity model
-toxic_classifier = pipeline("text-classification", model="unitary/toxic-bert", cache_dir="./")
+toxic_classifier = pipeline("text-classification", model="unitary/toxic-bert", cache_dir="./cache_dir")
 profanity.load_censor_words()
 custom_words = [
     "idiot", "moron", "dumb", "stupid", "loser", "bastard", "retard", "scumbag",