modified text sr to lang, test=doc

3 years ago · 395c923dee
parent 96abb33b5b
commit 395c923dee
1 changed files with 3 additions and 4 deletions
--- a/paddlespeech/cli/stats/infer.py
+++ b/paddlespeech/cli/stats/infer.py
@ -26,7 +26,7 @@ model_name_format = {
    'asr': 'Model-Language-Sample Rate',
    'cls': 'Model-Sample Rate',
    'st': 'Model-Source language-Target language',
-    'text': 'Model-Task-Sample Rate',
+    'text': 'Model-Task-Language',
    'tts': 'Model-Language'
 }

@ -180,8 +180,7 @@ class StatsExecutor():
                )
                self.show_support_models(pretrained_models)
            except BaseException:
-                print(
-                    "Failed to get the list of TEXT pretrained models.")
+                print("Failed to get the list of TEXT pretrained models.")

        elif self.task == 'tts':
            try: