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 branch 'main' of github.com:alibaba-damo-academy/FunASR

Browse Source 
merge
This commit is contained in:
游雁 2024-02-29 10:04:13 +08:00
commit 429bc77c03
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
funasr/auto/auto_model.py
View File

@ -392,7 +392,8 @@ class AutoModel:
# step.3 compute punc model # step.3 compute punc model
if self.punc_model is not None: if self.punc_model is not None:
if not len(result["text"]): if not len(result["text"]):
result['raw_text'] = '' if return_raw_text:
result['raw_text'] = ''
else: else:
self.punc_kwargs.update(cfg) self.punc_kwargs.update(cfg)
punc_res = self.inference(result["text"], model=self.punc_model, kwargs=self.punc_kwargs, **cfg) punc_res = self.inference(result["text"], model=self.punc_model, kwargs=self.punc_kwargs, **cfg)
@ -434,10 +435,13 @@ class AutoModel:
distribute_spk(sentence_list, sv_output) distribute_spk(sentence_list, sv_output)
result['sentence_info'] = sentence_list result['sentence_info'] = sentence_list
elif kwargs.get("sentence_timestamp", False): elif kwargs.get("sentence_timestamp", False):
sentence_list = timestamp_sentence(punc_res[0]['punc_array'], if not len(result['text']):
result['timestamp'], sentence_list = []
raw_text, else:
return_raw_text=return_raw_text) sentence_list = timestamp_sentence(punc_res[0]['punc_array'],
result['timestamp'],
raw_text,
return_raw_text=return_raw_text)
result['sentence_info'] = sentence_list result['sentence_info'] = sentence_list
if "spk_embedding" in result: del result['spk_embedding'] if "spk_embedding" in result: del result['spk_embedding']