Update app.py

app.py

@@ -8,16 +8,22 @@ import string
 import io
 import os
 import uvicorn
-
+import os
 # ✅ Set Hugging Face Cache Directory (Fixes Permission Error)
-CACHE_DIR = "/tmp/hf_cache"
-os.makedirs(CACHE_DIR, exist_ok=True)
+import os
+
+# ✅ Set Hugging Face Cache to a writable directory
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/tmp/hf_cache"
+
+
+
+
 
 # ✅ Initialize FastAPI
 app = FastAPI()
-
+pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector")
 # ✅ Load NSFW Image Classification Model (with custom cache directory)
-pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector", cache_dir="/tmp/hf_cache")
 
 
 # ✅ Load Toxic Text Classification Model