fangd123
2023-01-26 e10d648c85fd1cd174cffb47f6939d55a9bd83ba
better asr text result with json style output
1个文件已修改
6 ■■■■■ 已修改文件
funasr/bin/asr_inference_paraformer_vad_punc.py 6 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/bin/asr_inference_paraformer_vad_punc.py
@@ -3,6 +3,7 @@
import logging
import sys
import time
import json
from pathlib import Path
from typing import Optional
from typing import Sequence
@@ -637,8 +638,9 @@
                                                                                   postprocessed_result[2]
                        if len(word_lists) > 0: 
                            text_postprocessed_punc, punc_id_list = text2punc(word_lists, 20)
                            text_postprocessed_punc_time_stamp = "predictions: {}  time_stamp: {}".format(
                                text_postprocessed_punc, time_stamp_postprocessed)
                            text_postprocessed_punc_time_stamp = json.dumps({"predictions": text_postprocessed_punc,
                                                                             "time_stamp": time_stamp_postprocessed},
                                                                            ensure_ascii=False)
                        else:
                            text_postprocessed_punc = ""
                            punc_id_list = []