speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update unittest

Browse Source
This commit is contained in:
speech_asr 2023-03-10 18:55:22 +08:00
parent 0b7149931a
commit 2e5318a176
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_punctuation_pipeline.py
View File

@ -20,7 +20,7 @@ class TestTransformerInferencePipelines(unittest.TestCase):
)
inputs = "./egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vocab272727-pytorch/data/punc_example.txt"
rec_result = inference_pipeline(text_in=inputs)
logger.info("asr inference result: {0}".format(rec_result))
logger.info("punctuation inference result: {0}".format(rec_result))
def test_vadrealtime_inference_pipeline(self):
inference_pipeline = pipeline(
@ -36,7 +36,7 @@ class TestTransformerInferencePipelines(unittest.TestCase):
rec_result = inference_pipeline(text_in=vad, cache=cache_out)
cache_out = rec_result['cache']
rec_result_all += rec_result['text']
logger.info("asr inference result: {0}".format(rec_result_all))
logger.info("punctuation inference result: {0}".format(rec_result_all))
if __name__ == '__main__':