Update main.py

main.py

@@ -1,3 +1,11 @@
+import os
+
+# Redirect Hugging Face cache to local project directory (not /)
+os.environ["HF_HOME"] = "./hf_cache"
+os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
+os.environ["HF_DATASETS_CACHE"] = "./hf_cache"
+os.environ["HF_METRICS_CACHE"] = "./hf_cache"
+
 import uvicorn
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware