sagar008 commited on
Commit
7735398
·
verified ·
1 Parent(s): dfa7940

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -7,8 +7,11 @@ import uvicorn
7
  app = FastAPI()
8
 
9
  # Safe local cache directory for HF models
 
10
  os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
11
- os.environ["HF_HOME"] = "/app/hf"
 
 
12
 
13
  # Read Hugging Face token from environment
14
  HF_AUTH_TOKEN = os.getenv("HF_TOKEN")
 
7
  app = FastAPI()
8
 
9
  # Safe local cache directory for HF models
10
+ os.environ["HF_HOME"] = "/app/cache"
11
  os.environ["TRANSFORMERS_CACHE"] = "/app/cache"
12
+ os.environ["HF_DATASETS_CACHE"] = "/app/cache"
13
+ os.environ["XDG_CACHE_HOME"] = "/app/cache"
14
+
15
 
16
  # Read Hugging Face token from environment
17
  HF_AUTH_TOKEN = os.getenv("HF_TOKEN")