diff --git a/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py b/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py index baad18884..a6629cdd0 100644 --- a/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py +++ b/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py @@ -17,13 +17,10 @@ inference_pipeline = pipeline( ) 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) - #print(rec_result) - cache_out = rec_result['cache'] + rec_result = inference_pipeline(text_in=vad, param_dict=param_dict) rec_result_all += rec_result['text'] print(rec_result_all)