update infer.py

This commit is contained in:
mengzhe.cmz 2023-03-23 19:55:15 +08:00
parent 174053cfca
commit 111450d5e6

View File

@ -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)