speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #288 from alibaba-damo-academy/dev_gzf

Browse Source 
Dev gzf
This commit is contained in:
zhifu gao 2023-03-23 20:34:21 +08:00 committed by GitHub
commit 4e44c9f46e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/punctuation/espnet_model.py
View File

@ -24,8 +24,8 @@ class ESPnetPunctuationModel(AbsESPnetModel):
# ignore_id may be assumed as 0, shared with CTC-blank symbol for ASR.
self.ignore_id = ignore_id
if self.punc_model.with_vad():
print("This is a vad puncuation model.")
#if self.punc_model.with_vad():
# print("This is a vad puncuation model.")
def nll(
self,

20
funasr/runtime/python/websocket/ASR_server.py
View File

@ -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)