This commit is contained in:
speech_asr 2023-04-13 10:16:22 +08:00
parent 66fc9f0a9b
commit ee1fb9b722

View File

@ -26,16 +26,14 @@ class TestTransformerInferencePipelines(unittest.TestCase):
inference_pipeline = pipeline(
task=Tasks.punctuation,
model='damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727',
model_revision="v1.0.0",
)
inputs = "跨境河流是养育沿岸|人民的生命之源长期以来为帮助下游地区防灾减灾中方技术人员|在上游地区极为恶劣的自然条件下克服巨大困难甚至冒着生命危险|向印方提供汛期水文资料处理紧急事件中方重视印方在跨境河流问题上的关切|愿意进一步完善双方联合工作机制|凡是|中方能做的我们|都会去做而且会做得更好我请印度朋友们放心中国在上游的|任何开发利用都会经过科学|规划和论证兼顾上下游的利益"
vads = inputs.split("|")
cache_out = []
rec_result_all = "outputs:"
param_dict = {"cache": []}
for vad in vads:
rec_result = inference_pipeline(text_in=vad, cache=cache_out)
cache_out = rec_result['cache']
rec_result_all += rec_result['text']
rec_result = inference_pipeline(text_in=vad, param_dict=param_dict)
rec_result_all += rec_result["text"]
logger.info("punctuation inference result: {0}".format(rec_result_all))