Update app.py

app.py

@@ -6,37 +6,65 @@ from modelscope.hub.snapshot_download import snapshot_download
 
 # 1. 定义本地路径和远程仓库ID
 FUN_ASR_NANO_LOCAL_PATH = "./Fun-ASR/model"
-FUN_ASR_NANO_REPO_ID = "FunAudioLLM/Fun-ASR-Nano-2512"
 SENSE_VOICE_SMALL_LOCAL_PATH = "./Fun-ASR/model/SenseVoiceSmall"
-# SENSE_VOICE_SMALL_REPO_ID = "FunAudioLLM/SenseVoiceSmall"
-# REPO_TYPE = "hf"  # "hf" for Hugging Face, "ms" for ModelScope
-SENSE_VOICE_SMALL_REPO_ID = "iic/SenseVoiceSmall"
-REPO_TYPE = "ms"  
 
-# 2. 检查本地是否存在，不存在则下载
-if not os.path.exists(FUN_ASR_NANO_LOCAL_PATH):
-    from modelscope import HubApi
-    api= HubApi()
-    api.login(os.getenv("MODELSCOPE_TOKEN"))
-    print(f"正在下载模型 Fun-ASR-Nano 到 {FUN_ASR_NANO_LOCAL_PATH} ...")
-    snapshot_download(
-        repo_id=FUN_ASR_NANO_REPO_ID,
-        local_dir=FUN_ASR_NANO_LOCAL_PATH,
-        ignore_patterns=["*.onnx"], # 如果你不需要onnx文件，可以过滤掉以节省时间和空间
-    )
+# 配置仓库类型 (通过修改这个变量来切换平台)
+# "ms" = ModelScope, "hf" = Hugging Face
+REPO_TYPE = "hf"  # 修改这里来切换平台
+
+# 根据仓库类型配置仓库ID
+if REPO_TYPE == "hf":
+    # Hugging Face 配置
+    FUN_ASR_NANO_REPO_ID = "FunAudioLLM/Fun-ASR-Nano-2512"
+    SENSE_VOICE_SMALL_REPO_ID = "FunAudioLLM/SenseVoiceSmall"
+else:  # "ms"
+    # ModelScope 配置
+    FUN_ASR_NANO_REPO_ID = "FunAudioLLM/Fun-ASR-Nano-2512"
+    SENSE_VOICE_SMALL_REPO_ID = "iic/SenseVoiceSmall" 
+
+# 2. 导入下载库
+if REPO_TYPE == "hf":
+    from huggingface_hub import snapshot_download as hf_snapshot_download
+else:
+    from modelscope.hub.snapshot_download import snapshot_download as ms_snapshot_download
+
+def download_model(repo_id, local_path):
+    """
+    根据平台类型下载模型
+
+    Args:
+        repo_id (str): 仓库ID
+        local_path (str): 本地存储路径
+    """
+    if REPO_TYPE == "hf":
+        # Hugging Face 下载逻辑
+        print(f"正在从 Hugging Face 下载模型 {repo_id} 到 {local_path} ...")
+        hf_snapshot_download(
+            repo_id=repo_id,
+            local_dir=local_path,
+            ignore_patterns=["*.onnx"], # 如果你不需要onnx文件，可以过滤掉以节省时间和空间
+        )
+    else:  # "ms"
+        # ModelScope 下载逻辑
+        print(f"正在从 ModelScope 下载模型 {repo_id} 到 {local_path} ...")
+        from modelscope import HubApi
+        api = HubApi()
+        api.login(os.getenv("MODELSCOPE_TOKEN"))
+        ms_snapshot_download(
+            repo_id=repo_id,
+            local_dir=local_path,
+            ignore_patterns=["*.onnx"], # 如果你不需要onnx文件，可以过滤掉以节省时间和空间
+        )
     print("模型下载完毕！")
+
+# 3. 检查本地是否存在，不存在则下载
+if not os.path.exists(FUN_ASR_NANO_LOCAL_PATH):
+    download_model(FUN_ASR_NANO_REPO_ID, FUN_ASR_NANO_LOCAL_PATH)
 else:
     print("检测到本地模型文件，跳过下载。")
 
-
 if not os.path.exists(SENSE_VOICE_SMALL_LOCAL_PATH):
-    print(f"正在下载模型 {SENSE_VOICE_SMALL_REPO_ID} 到 {SENSE_VOICE_SMALL_LOCAL_PATH} ...")
-    snapshot_download(
-        repo_id=SENSE_VOICE_SMALL_REPO_ID,
-        local_dir=SENSE_VOICE_SMALL_LOCAL_PATH,
-        ignore_patterns=["*.onnx"], # 如果你不需要onnx文件，可以过滤掉以节省时间和空间
-    )
-    print("模型下载完毕！")
+    download_model(SENSE_VOICE_SMALL_REPO_ID, SENSE_VOICE_SMALL_LOCAL_PATH)
 else:
     print("检测到本地模型文件，跳过下载。")