From dfcc5d47587d3e793cbfec2e9509c0e9a9e1732c Mon Sep 17 00:00:00 2001
From: Dogvane Huang <dogvane@gmail.com>
Date: 星期二, 02 七月 2024 12:24:13 +0800
Subject: [PATCH] fix c# demo project to new onnx model files (#1689)
---
tests/test_lm_pipeline.py | 6 ++++--
1 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/tests/test_lm_pipeline.py b/tests/test_lm_pipeline.py
index 3a5ec57..f67c9a5 100644
--- a/tests/test_lm_pipeline.py
+++ b/tests/test_lm_pipeline.py
@@ -6,20 +6,22 @@
logger = get_logger()
+
class TestTransformerInferencePipelines(unittest.TestCase):
def test_funasr_path(self):
import funasr
import os
+
logger.info("run_dir:{0} ; funasr_path: {1}".format(os.getcwd(), funasr.__file__))
def test_inference_pipeline(self):
inference_pipeline = pipeline(
task=Tasks.language_score_prediction,
- model='damo/speech_transformer_lm_zh-cn-common-vocab8404-pytorch',
+ model="damo/speech_transformer_lm_zh-cn-common-vocab8404-pytorch",
)
rec_result = inference_pipeline(text_in="hello 澶� 瀹� 濂� 鍛�")
logger.info("lm inference result: {0}".format(rec_result))
-if __name__ == '__main__':
+if __name__ == "__main__":
unittest.main()
--
Gitblit v1.9.1