Browse Source

ai_asr.py 中把 model_name 抽离出来作为变量

reghao 4 weeks ago
parent
commit
281f572dff
1 changed files with 6 additions and 4 deletions
  1. 6 4
      service/ai_asr.py

+ 6 - 4
service/ai_asr.py

@@ -11,6 +11,9 @@ from funasr import AutoModel
 logger = logging.getLogger(__name__)
 asr_model: Optional[AutoModel] = None
 is_model_ready = False
+model_name1 = 'paraformer-zh'
+model_name2 = 'fsmn-vad'
+model_name3 = 'ct-punc'
 
 
 def get_asr_model():
@@ -39,13 +42,12 @@ async def init_funasr():
         # 使用 run_in_executor 避免阻塞主事件循环
         loop = asyncio.get_event_loop()
 
-        # 定义具体的加载逻辑
         def load():
             return AutoModel(
-                model="paraformer-zh",
-                vad_model="fsmn-vad",
+                model=model_name1,
+                vad_model=model_name2,
                 vad_kwargs={"max_single_segment_time": 30000},
-                punc_model="ct-punc",
+                punc_model=model_name3,
                 device="cuda:0" if torch.cuda.is_available() else "cpu",
                 disable_update=True
             )