From 6464447d074ed00d872aee5dd4a9726996e561c3 Mon Sep 17 00:00:00 2001
From: Xian Shi <40013335+R1ckShi@users.noreply.github.com>
Date: 星期三, 28 六月 2023 15:52:14 +0800
Subject: [PATCH] update tp unitest (#680)
---
tests/test_tp_pipeline.py | 29 +++++++++++++++++++++++++++++
egs_modelscope/tp/speech_timestamp_prediction-v1-16k-offline/demo.py | 3 +--
2 files changed, 30 insertions(+), 2 deletions(-)
diff --git a/egs_modelscope/tp/speech_timestamp_prediction-v1-16k-offline/demo.py b/egs_modelscope/tp/speech_timestamp_prediction-v1-16k-offline/demo.py
index 3116f6d..581f7aa 100644
--- a/egs_modelscope/tp/speech_timestamp_prediction-v1-16k-offline/demo.py
+++ b/egs_modelscope/tp/speech_timestamp_prediction-v1-16k-offline/demo.py
@@ -4,8 +4,7 @@
inference_pipeline = pipeline(
task=Tasks.speech_timestamp,
model='damo/speech_timestamp_prediction-v1-16k-offline',
- model_revision='v1.1.0',
- output_dir=None)
+ model_revision='v1.1.0')
rec_result = inference_pipeline(
audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_timestamps.wav',
diff --git a/tests/test_tp_pipeline.py b/tests/test_tp_pipeline.py
new file mode 100644
index 0000000..21284a6
--- /dev/null
+++ b/tests/test_tp_pipeline.py
@@ -0,0 +1,29 @@
+import unittest
+
+from modelscope.pipelines import pipeline
+from modelscope.utils.constant import Tasks
+from modelscope.utils.logger import get_logger
+
+logger = get_logger()
+
+class TestTimestampPredictionPipelines(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.speech_timestamp,
+ model='damo/speech_timestamp_prediction-v1-16k-offline',
+ model_revision='v1.1.0')
+
+ rec_result = inference_pipeline(
+ audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_timestamps.wav',
+ text_in='涓� 涓� 涓� 澶� 骞� 娲� 鍥� 瀹� 涓� 浠� 涔� 璺� 鍒� 瑗� 澶� 骞� 娲� 鏉� 浜� 鍛�',)
+ print(rec_result)
+ logger.info("punctuation inference result: {0}".format(rec_result))
+
+
+if __name__ == '__main__':
+ unittest.main()
\ No newline at end of file
--
Gitblit v1.9.1