add ASR_Model Select

tools/asr/config.py

@@ -0,0 +1,31 @@
+import os
+
+def check_fw_local_models():
+    '''
+    启动时检查本地是否有 Faster Whisper 模型.
+    '''
+    model_size_list = [
+        "tiny",     "tiny.en", 
+        "base",     "base.en", 
+        "small",    "small.en", 
+        "medium",   "medium.en", 
+        "large",    "large-v1", 
+        "large-v2", "large-v3"]
+    for i, size in enumerate(model_size_list):
+        if os.path.exists(f'tools/asr/models/faster-whisper-{size}'):
+            model_size_list[i] = size + '(local)'
+    return model_size_list
+
+asr_dict = {
+    "达摩 ASR (中文)": {
+        'lang': ['zh'],
+        'size': ['large'],
+        'path': 'funasr_asr.py',
+    },
+    "Faster Whisper (多语种)": {
+        'lang': ['auto', 'zh', 'en', 'ja'],
+        'size': check_fw_local_models(),
+        'path': 'fasterwhisper_asr.py'
+    }
+}
+