speech_asr 3 anni fa
parent
commit
8552131d86
1 ha cambiato i file con 26 aggiunte e 0 eliminazioni
  1. 26 0
      tests/test_asr_inference_pipeline.py

+ 26 - 0
tests/test_asr_inference_pipeline.py

@@ -142,6 +142,7 @@ class TestParaformerInferencePipelines(unittest.TestCase):
             audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_zh.wav')
         logger.info("asr inference result: {0}".format(rec_result))
 
+
 class TestParaformerBertInferencePipelines(unittest.TestCase):
     def test_funasr_path(self):
         import funasr
@@ -165,5 +166,30 @@ class TestParaformerBertInferencePipelines(unittest.TestCase):
         logger.info("asr inference result: {0}".format(rec_result))
 
 
+class TestUniasrInferencePipelines(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_uniasr_2pass_cantonese_chs_16k_common_offline(self):
+        inference_pipeline = pipeline(
+            task=Tasks.auto_speech_recognition,
+            model='damo/speech_UniASR_asr_2pass-cantonese-CHS-16k-common-vocab1468-tensorflow1-online')
+        rec_result = inference_pipeline(
+            audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_cantonese-CHS.wav',
+            param_dict={"decoding_model": "offline"})
+        logger.info("asr inference result: {0}".format(rec_result))
+
+    def test_uniasr_2pass_cantonese_chs_16k_common_online(self):
+        inference_pipeline = pipeline(
+            task=Tasks.auto_speech_recognition,
+            model='damo/speech_UniASR_asr_2pass-cantonese-CHS-16k-common-vocab1468-tensorflow1-online')
+        rec_result = inference_pipeline(
+            audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/asr_example_cantonese-CHS.wav',
+            param_dict={"decoding_model":"normal"})
+        logger.info("asr inference result: {0}".format(rec_result))
+
+
 if __name__ == '__main__':
     unittest.main()