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

update for funasr-app and bug fix

Browse Source
This commit is contained in:
shixian.shi 2024-02-06 17:38:58 +08:00
parent 14a9e01781
commit 78ee057a99
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
funasr/auto/auto_model.py
View File

@ -385,11 +385,11 @@ class AutoModel:
result["text"] = punc_res[0]["text"]
# speaker embedding cluster after resorted
if self.spk_model is not None:
if self.spk_model is not None and kwargs.get('return_spk_res', True):
all_segments = sorted(all_segments, key=lambda x: x[0])
spk_embedding = result['spk_embedding']
labels = self.cb_model(spk_embedding.cpu(), oracle_num=kwargs['preset_spk_num'])
del result['spk_embedding']
labels = self.cb_model(spk_embedding.cpu(), oracle_num=kwargs.get('preset_spk_num', None))
# del result['spk_embedding']
sv_output = postprocess(all_segments, None, labels, spk_embedding.cpu())
if self.spk_mode == 'vad_segment': # recover sentence_list
sentence_list = []
@ -409,6 +409,7 @@ class AutoModel:
result['timestamp'], \
result['raw_text'])
result['sentence_info'] = sentence_list
del result['spk_embedding']
result["key"] = key
results_ret_list.append(result)