Update app.py

app.py

@@ -17,7 +17,8 @@ os.makedirs(CACHE_DIR, exist_ok=True)
 app = FastAPI()
 
 # ✅ Load NSFW Image Classification Model (with custom cache directory)
-pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector", cache_dir=CACHE_DIR)
+pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector", cache_dir="/tmp/hf_cache")
+
 
 # ✅ Load Toxic Text Classification Model
 try: