From 7fec275c7e75aefc7d72f4ab287ddb4fc52ec93f Mon Sep 17 00:00:00 2001
From: tianhao zhang <15600919271@163.com>
Date: Mon, 28 Nov 2022 16:29:51 +0000
Subject: [PATCH] support wav2vec2-zh cli, test=asr

---
 paddlespeech/__init__.py      | 4 ++++
 paddlespeech/cli/ssl/infer.py | 4 ++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/paddlespeech/__init__.py b/paddlespeech/__init__.py
index b781c4a8e..29db45cc5 100644
--- a/paddlespeech/__init__.py
+++ b/paddlespeech/__init__.py
@@ -14,3 +14,7 @@
 import _locale
 
 _locale._getdefaultlocale = (lambda *args: ['en_US', 'utf8'])
+
+
+
+
diff --git a/paddlespeech/cli/ssl/infer.py b/paddlespeech/cli/ssl/infer.py
index 36e577110..ca0f5bf7d 100644
--- 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")