Update main.py

main.py

@@ -1,3 +1,11 @@
+import os
+# 2️⃣ Optional: force cache to writable directory
+os.environ["HF_HOME"] = "/tmp/hf"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf"
+os.environ["HF_DATASETS_CACHE"] = "/tmp/hf"
+os.makedirs("/tmp/hf", exist_ok=True)
+
+
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM