ynsbyrm commited on
Commit
834a493
·
verified ·
1 Parent(s): 9321d26

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +4 -3
app/main.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
- os.environ["HF_HOME"] = "/data/.cache"
3
- os.environ["TRANSFORMERS_CACHE"] = "/data/.cache"
4
- os.environ["HF_DATASETS_CACHE"] = "/data/.cache"
 
5
 
6
  from fastapi import FastAPI
7
  from pydantic import BaseModel
 
1
  import os
2
+ os.environ["HF_HOME"] = "/home/appuser/.cache/huggingface"
3
+ os.environ["TRANSFORMERS_CACHE"] = "/home/appuser/.cache/huggingface"
4
+ os.environ["HUGGINGFACE_HUB_CACHE"] = "/home/appuser/.cache/huggingface"
5
+
6
 
7
  from fastapi import FastAPI
8
  from pydantic import BaseModel