优化

app.py

@@ -39,10 +39,17 @@ spk = None
 debug = False
 MODEL_REPO = 'Asankilp/gochiusa-sovits4.0'
 MODEL_NAME = 'G_240000.pth'
+DOWNLOAD_DIR = './downloaded'
 CHECKPOINT_REPO = 'WitchHuntTV/checkpoint_best_legacy_500.pt'
-model_pth = hf_hub_download(repo_id=MODEL_REPO,filename=MODEL_NAME)
-model_configfile = hf_hub_download(repo_id=MODEL_REPO,filename='config.json')
-checkpoint_pt = hf_hub_download(repo_id=CHECKPOINT_REPO,filename='checkpoint_best_legacy_500.pt',local_dir='./pretrain')
+if not os.path.exists(os.path.join(DOWNLOAD_DIR,MODEL_NAME):
+    print('下载模型文件...')
+    model_pth = hf_hub_download(repo_id=MODEL_REPO,filename=MODEL_NAME,local_dir=DOWNLOAD_DIR)
+if not os.path.exists(os.path.join(DOWNLOAD_DIR,'config.json'):
+    print('下载配置文件...')
+    model_configfile = hf_hub_download(repo_id=MODEL_REPO,filename='config.json',local_dir=DOWNLOAD_DIR)
+if not os.path.exists(os.path.join('./pretrain/checkpoint_best_legacy_500.pt'):
+    print('下载预训练模型文件...)
+    checkpoint_pt = hf_hub_download(repo_id=CHECKPOINT_REPO,filename='checkpoint_best_legacy_500.pt',local_dir='./pretrain')
 class HParams():
     def __init__(self, **kwargs):
         for k, v in kwargs.items():
@@ -182,8 +189,8 @@ models_info = [
                       请问你今天要来点so-vits-svc 4.0模型吗？\n\n
                       * 模型尚处于半成品（
                        """,
-        "model_path": model_pth,
-        "config_path": model_configfile,
+        "model_path": os.path.join(DOWNLOAD_DIR,MODEL_NAME),
+        "config_path": os.path.join(DOWNLOAD_DIR,'config.json'),
     }
 ]