Spaces:
Running
on
Zero
Running
on
Zero
James Zhou
commited on
Commit
·
0b821d0
1
Parent(s):
4ab7964
[update] app
Browse files
app.py
CHANGED
|
@@ -78,7 +78,7 @@ def auto_load_models() -> str:
|
|
| 78 |
# 如果模型路径不存在,尝试从HuggingFace下载
|
| 79 |
if not os.path.exists(MODEL_PATH):
|
| 80 |
logger.info(f"模型路径 {MODEL_PATH} 不存在,开始从HuggingFace下载...")
|
| 81 |
-
download_result = download_model_from_hf(local_dir=MODEL_PATH.rstrip('/'))
|
| 82 |
if "失败" in download_result:
|
| 83 |
return download_result
|
| 84 |
|
|
@@ -89,7 +89,7 @@ def auto_load_models() -> str:
|
|
| 89 |
if CONFIG_PATH.startswith("configs/"):
|
| 90 |
config_dir = os.path.dirname(CONFIG_PATH)
|
| 91 |
if not os.path.exists(config_dir):
|
| 92 |
-
download_result = download_model_from_hf(local_dir="./")
|
| 93 |
if "失败" in download_result:
|
| 94 |
return download_result
|
| 95 |
|
|
|
|
| 78 |
# 如果模型路径不存在,尝试从HuggingFace下载
|
| 79 |
if not os.path.exists(MODEL_PATH):
|
| 80 |
logger.info(f"模型路径 {MODEL_PATH} 不存在,开始从HuggingFace下载...")
|
| 81 |
+
download_result = download_model_from_hf(repo_id="jogin/test_pub_models", local_dir=MODEL_PATH.rstrip('/'))
|
| 82 |
if "失败" in download_result:
|
| 83 |
return download_result
|
| 84 |
|
|
|
|
| 89 |
if CONFIG_PATH.startswith("configs/"):
|
| 90 |
config_dir = os.path.dirname(CONFIG_PATH)
|
| 91 |
if not os.path.exists(config_dir):
|
| 92 |
+
download_result = download_model_from_hf(repo_id="jogin/test_pub_models", local_dir="./")
|
| 93 |
if "失败" in download_result:
|
| 94 |
return download_result
|
| 95 |
|