Update inference_webui.py

inference_webui.py

@@ -90,6 +90,42 @@ import random
 
 from module.models import Generator, SynthesizerTrn, SynthesizerTrnV3
 
+from huggingface_hub import hf_hub_download
+
+# 1. 定义仓库信息和本地目标路径
+# ----------------------------------------------------
+# 您的远程模型仓库
+repo_id = "l73jiang/Seia-GPT-SOVITS-ProPlus"  # <-- 请替换成您的用户名和仓库名
+# 您希望文件被存放在 Space 中的哪个文件夹
+target_dir = "pretrained_models"
+
+# 2. 将所有需要下载的模型文件名放入一个列表
+# ----------------------------------------------------
+files_to_download = [
+    "Seia-e15.ckpt",
+    "Seia_e8_s240.pth"  # <-- 新增了第二个模型文件
+]
+
+# 3. 确保目标文件夹存在（这个操作只需执行一次）
+# ----------------------------------------------------
+os.makedirs(target_dir, exist_ok=True)
+print(f"目标文件夹 '{target_dir}' 已准备就绪。")
+
+# 4. 循环遍历列表，下载每一个文件
+# ----------------------------------------------------
+for filename in files_to_download:
+    print(f"-> 开始从仓库 '{repo_id}' 下载 '{filename}'...")
+    try:
+        hf_hub_download(
+            repo_id=repo_id,
+            filename=filename,
+            local_dir=target_dir,  # 所有文件都下载到同一个目标文件夹
+            local_dir_use_symlinks=False
+        )
+        print(f"   文件 '{filename}' 下载成功！")
+    except Exception as e:
+        # 增加一个错误处理，这样如果某个文件下载失败，应用不会直接崩溃
+        print(f"   !!! 下载文件 '{filename}' 时发生错误: {e}")
 
 def set_seed(seed):
     if seed == -1: