Merge pull request #44 from fangd123/dev

better asr text result with json style output
This commit is contained in:
zhifu gao 2023-01-26 14:06:12 +08:00 committed by GitHub
commit 96d637a58b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,6 +3,7 @@ import argparse
import logging
import sys
import time
import json
from pathlib import Path
from typing import Optional
from typing import Sequence
@ -637,8 +638,9 @@ def inference_modelscope(
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 = []