Kushagra13 commited on
Commit
07c2738
·
verified ·
1 Parent(s): c96dbfa

Update src/pages/prt111.py

Browse files
Files changed (1) hide show
  1. 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
- os.environ["HF_HOME"] = "/temp/huggingface"
22
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
23
- os.environ["HF_DATASETS_CACHE"] = "/tmp/huggingface"
 
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