From 348673947693a7e02f74df84f5eaa5dd2579d501 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Thu, 23 Mar 2023 20:33:03 +0800 Subject: [PATCH] websocket --- funasr/runtime/python/websocket/ASR_server.py | 20 ++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/funasr/runtime/python/websocket/ASR_server.py b/funasr/runtime/python/websocket/ASR_server.py index 17f73bb54..9d0fd0b10 100644 --- a/funasr/runtime/python/websocket/ASR_server.py +++ b/funasr/runtime/python/websocket/ASR_server.py @@ -70,14 +70,16 @@ inference_pipeline_asr = pipeline( param_dict=param_dict_asr, ngpu=args.ngpu, ) - -param_dict_punc = {'cache': list()} -inference_pipeline_punc = pipeline( - task=Tasks.punctuation, - model=args.punc_model, - model_revision=None, - ngpu=args.ngpu, -) +if args.punc_model is not None: + param_dict_punc = {'cache': list()} + inference_pipeline_punc = pipeline( + task=Tasks.punctuation, + model=args.punc_model, + model_revision=None, + ngpu=args.ngpu, + ) +else: + inference_pipeline_punc = None print("model loaded") @@ -124,7 +126,7 @@ def asr(): # 推理 speek.task_done() if len(audio_in) > 0: rec_result = inference_pipeline_asr(audio_in=audio_in) - if 'text' in rec_result: + if inference_pipeline_punc is not None and 'text' in rec_result: rec_result = inference_pipeline_punc(text_in=rec_result['text'], param_dict=param_dict_punc) print(rec_result["text"]) time.sleep(0.1)