James Zhou commited on
Commit
3fff971
·
1 Parent(s): 50707ba

[update] app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -79,7 +79,7 @@ def auto_load_models() -> str:
79
  # 如果模型路径不存在,尝试从HuggingFace下载
80
  if not os.path.exists(MODEL_PATH):
81
  logger.info(f"模型路径 {MODEL_PATH} 不存在,开始从HuggingFace下载...")
82
- download_result = download_model_from_hf(repo_id="jogin/test_pub_models", local_dir=MODEL_PATH.rstrip('/'))
83
  if "失败" in download_result:
84
  return download_result
85
 
@@ -90,7 +90,7 @@ def auto_load_models() -> str:
90
  if CONFIG_PATH.startswith("configs/"):
91
  config_dir = os.path.dirname(CONFIG_PATH)
92
  if not os.path.exists(config_dir):
93
- download_result = download_model_from_hf(repo_id="jogin/test_pub_models", local_dir="./")
94
  if "失败" in download_result:
95
  return download_result
96
 
 
79
  # 如果模型路径不存在,尝试从HuggingFace下载
80
  if not os.path.exists(MODEL_PATH):
81
  logger.info(f"模型路径 {MODEL_PATH} 不存在,开始从HuggingFace下载...")
82
+ download_result = download_model_from_hf(local_dir=MODEL_PATH.rstrip('/'))
83
  if "失败" in download_result:
84
  return download_result
85
 
 
90
  if CONFIG_PATH.startswith("configs/"):
91
  config_dir = os.path.dirname(CONFIG_PATH)
92
  if not os.path.exists(config_dir):
93
+ download_result = download_model_from_hf(local_dir="./")
94
  if "失败" in download_result:
95
  return download_result
96