cao commited on
Commit
9ad4750
·
1 Parent(s): d249ad8
Files changed (1) hide show
  1. src/streamlit_app.py +7 -0
src/streamlit_app.py CHANGED
@@ -18,6 +18,13 @@ Upload a **CSV** file with columns `Peptide` and `HLA`,
18
  or a **FASTA** file containing peptide sequences (headers optionally include HLA type).
19
  """)
20
 
 
 
 
 
 
 
 
21
  # ==============================
22
  # 模型加载函数(缓存)
23
  # ==============================
 
18
  or a **FASTA** file containing peptide sequences (headers optionally include HLA type).
19
  """)
20
 
21
+ import os
22
+ os.environ["HF_HOME"] = "/data/huggingface"
23
+ os.environ["TRANSFORMERS_CACHE"] = "/data/huggingface"
24
+ os.environ["TORCH_HOME"] = "/data/huggingface"
25
+ os.environ["ESM_CACHE_DIR"] = "/data/phla_cache"
26
+ os.makedirs("/data/phla_cache", exist_ok=True)
27
+
28
  # ==============================
29
  # 模型加载函数(缓存)
30
  # ==============================