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 raw_text

Browse Source
This commit is contained in:
shixian.shi 2024-01-22 18:44:07 +08:00
parent eed0c23816
commit 0ee4565c8e
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/auto/auto_model.py
View File

@ -406,7 +406,7 @@ class AutoModel:
for res, vadsegment in zip(restored_data, vadsegments): for res, vadsegment in zip(restored_data, vadsegments):
sentence_list.append({"start": vadsegment[0],\ sentence_list.append({"start": vadsegment[0],\
"end": vadsegment[1], "end": vadsegment[1],
"sentence": res['text'], "sentence": res['raw_text'],
"timestamp": res['timestamp']}) "timestamp": res['timestamp']})
else: # punc_segment else: # punc_segment
sentence_list = timestamp_sentence(punc_res[0]['punc_array'], \ sentence_list = timestamp_sentence(punc_res[0]['punc_array'], \

2
funasr/models/seaco_paraformer/model.py
View File

@ -420,7 +420,7 @@ class SeacoParaformer(BiCifParaformer, Paraformer):
if ibest_writer is not None: if ibest_writer is not None:
ibest_writer["token"][key[i]] = " ".join(token) ibest_writer["token"][key[i]] = " ".join(token)
# ibest_writer["text"][key[i]] = text ibest_writer["raw_text"][key[i]] = text
ibest_writer["timestamp"][key[i]] = time_stamp_postprocessed ibest_writer["timestamp"][key[i]] = time_stamp_postprocessed
ibest_writer["text"][key[i]] = text_postprocessed ibest_writer["text"][key[i]] = text_postprocessed
else: else: