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 auto_model.py

Browse Source 
修复空字串进入speaker model时报raw_text变量不存在的bug
This commit is contained in:
liugz18 2024-06-12 12:30:00 +08:00 committed by GitHub
parent 1300d38bf0
commit 648c03fdbb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
funasr/auto/auto_model.py
View File

@ -469,10 +469,11 @@ class AutoModel:
return_raw_text = kwargs.get("return_raw_text", False) return_raw_text = kwargs.get("return_raw_text", False)
# step.3 compute punc model # step.3 compute punc model
raw_text = None
if self.punc_model is not None: if self.punc_model is not None:
if not len(result["text"].strip()): if not len(result["text"].strip()):
if return_raw_text: if return_raw_text:
result["raw_text"] = "" result["raw_text"] = raw_text = ""
else: else:
deep_update(self.punc_kwargs, cfg) deep_update(self.punc_kwargs, cfg)
punc_res = self.inference( punc_res = self.inference(
@ -482,9 +483,7 @@ class AutoModel:
if return_raw_text: if return_raw_text:
result["raw_text"] = raw_text result["raw_text"] = raw_text
result["text"] = punc_res[0]["text"] result["text"] = punc_res[0]["text"]
else:
raw_text = None
# speaker embedding cluster after resorted # speaker embedding cluster after resorted
if self.spk_model is not None and kwargs.get("return_spk_res", True): if self.spk_model is not None and kwargs.get("return_spk_res", True):
if raw_text is None: if raw_text is None: