support wav2vec2-zh cli, test=asr

3 years ago · 7fec275c7e
parent d9f28a87d4
commit 7fec275c7e
2 changed files with 6 additions and 2 deletions
--- a/paddlespeech/init.py
+++ b/paddlespeech/init.py
@ -14,3 +14,7 @@
 import _locale

 _locale._getdefaultlocale = (lambda *args: ['en_US', 'utf8'])
+
+
+
+
--- a/paddlespeech/cli/ssl/infer.py
+++ b/paddlespeech/cli/ssl/infer.py
@ -138,9 +138,9 @@ class SSLExecutor(BaseExecutor):

        if model_type is None:
            if lang == 'en':
-                model_type = wav2vec2ASR_librispeech
+                model_type = 'wav2vec2ASR_librispeech'
            elif lang == 'zh':
-                model_type = wav2vec2ASR_aishell1
+                model_type = 'wav2vec2ASR_aishell1'
            else:
                logger.error(
                    "invalid lang, please input --lang en or --lang zh")