Fix cache directory permission issue with explicit cache_dir

app.py

@@ -9,6 +9,7 @@ import uvicorn
 cache_dir = "/tmp/hf_cache"
 os.environ["HF_HOME"] = cache_dir
 os.environ["HUGGINGFACE_HUB_CACHE"] = cache_dir
+os.environ["TRANSFORMERS_CACHE"] = cache_dir  # Add this for backward compatibility
 
 # Create the cache directory if it doesn't exist
 if not os.path.exists(cache_dir):
@@ -17,8 +18,12 @@ if not os.path.exists(cache_dir):
 app = FastAPI()
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
-# Load the zero-shot classification model
-classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+# Load the zero-shot classification model with explicit cache directory
+classifier = pipeline(
+    "zero-shot-classification",
+    model="facebook/bart-large-mnli",
+    cache_dir=cache_dir
+)
 
 @app.get("/")
 async def index():