Update app.py

app.py

@@ -11,16 +11,17 @@ from flask import Flask, request, jsonify
 from transformers import pipeline
 from PIL import Image
 
-# ✅ Set Hugging Face cache directory to a writable temp path
+# ✅ Set Hugging Face cache to a writable directory
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
+os.environ["HF_HOME"] = "/tmp/huggingface_cache"
+
+# ✅ Ensure the cache directory exists
+os.makedirs("/tmp/huggingface_cache", exist_ok=True)
 
 # ✅ Initialize FastAPI and Flask
 app_fastapi = FastAPI()
 app_flask = Flask(__name__)
 
-# ✅ Ensure the cache directory exists (use /tmp for permission safety)
-os.makedirs("/tmp/huggingface_cache", exist_ok=True)
-
 # ✅ Load NSFW Image Classification Model
 pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector")