Update main.py

main.py

@@ -1,11 +1,12 @@
 import os
+# ✅ Set cache to writable directory
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+os.environ["HF_HOME"] = "/tmp/hf-home"
+
 from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from transformers import pipeline
 
-# ✅ Set cache to writable directory
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
-
 app = FastAPI()
 
 # Load model once when the app starts