Spaces:
Running
Running
设置环境变量以指定 Hugging Face 缓存路径
Browse files- blkeras.py +2 -1
blkeras.py
CHANGED
@@ -23,7 +23,8 @@ from RequestModel import PredictRequest
|
|
23 |
from app import TextRequest
|
24 |
from us_stock import find_stock_codes_or_names
|
25 |
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
26 |
-
|
|
|
27 |
|
28 |
# 加载模型
|
29 |
model = None
|
|
|
23 |
from app import TextRequest
|
24 |
from us_stock import find_stock_codes_or_names
|
25 |
os.environ["TOKENIZERS_PARALLELISM"] = "false"
|
26 |
+
# 设置环境变量,指定 Hugging Face 缓存路径
|
27 |
+
os.environ["HF_HOME"] = "/tmp/huggingface"
|
28 |
|
29 |
# 加载模型
|
30 |
model = None
|