James Zhou commited on
Commit
0b821d0
·
1 Parent(s): 4ab7964

[update] app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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