Spaces:
Sleeping
Sleeping
Update src/pages/prt111.py
Browse files- src/pages/prt111.py +4 -3
src/pages/prt111.py
CHANGED
@@ -18,9 +18,10 @@ import networkx as nx
|
|
18 |
from sklearn.manifold import TSNE
|
19 |
from dotenv import load_dotenv
|
20 |
|
21 |
-
|
22 |
-
os.environ["TRANSFORMERS_CACHE"] = "/tmp/
|
23 |
-
os.environ["
|
|
|
24 |
|
25 |
load_dotenv() # load .env file
|
26 |
|
|
|
18 |
from sklearn.manifold import TSNE
|
19 |
from dotenv import load_dotenv
|
20 |
|
21 |
+
# Use /tmp/hf_cache for model files (fixes permission errors on cloud hosting)
|
22 |
+
os.environ["TRANSFORMERS_CACHE"] = os.getenv("TRANSFORMERS_CACHE", "/tmp/hf_cache")
|
23 |
+
os.environ["HF_HOME"] = os.getenv("HF_HOME", "/tmp/huggingface")
|
24 |
+
os.environ["HF_DATASETS_CACHE"] = os.getenv("HF_DATASETS_CACHE", "/tmp/huggingface")
|
25 |
|
26 |
load_dotenv() # load .env file
|
27 |
|