From 3a7896fc9693a1467f7b9a9b20d48168893f47d6 Mon Sep 17 00:00:00 2001
From: root <root@yq01-sys-hic-k8s-v100-qm-a225-0004.yq01.baidu.com>
Date: Fri, 6 May 2022 06:56:57 +0000
Subject: [PATCH] update cli, test=asr

---
 paddlespeech/cli/asr/pretrained_models.py | 16 +++++++++++++---
 tests/unit/cli/test_cli.sh                |  1 +
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/paddlespeech/cli/asr/pretrained_models.py b/paddlespeech/cli/asr/pretrained_models.py
index 80b04aa4..7f198ad6 100644
--- a/paddlespeech/cli/asr/pretrained_models.py
+++ b/paddlespeech/cli/asr/pretrained_models.py
@@ -27,6 +27,16 @@ pretrained_models = {
         'ckpt_path':
         'exp/conformer/checkpoints/wenetspeech',
     },
+   "conformer_online_wenetspeech-zh-16k": {
+        'url':
+        'https://paddlespeech.bj.bcebos.com/s2t/wenetspeech/asr1/asr1_chunk_conformer_wenetspeech_ckpt_1.0.0a.model.tar.gz',
+        'md5':
+        'b8c02632b04da34aca88459835be54a6',
+        'cfg_path':
+        'model.yaml',
+        'ckpt_path':
+        'exp/chunk_conformer/checkpoints/avg_10',
+    },
     "conformer_online_multicn-zh-16k": {
         'url':
         'https://paddlespeech.bj.bcebos.com/s2t/multi_cn/asr1/asr1_chunk_conformer_multi_cn_ckpt_0.2.0.model.tar.gz',
@@ -69,13 +79,13 @@ pretrained_models = {
     },
     "deepspeech2online_wenetspeech-zh-16k": {
         'url':
-        'https://paddlespeech.bj.bcebos.com/s2t/wenetspeech/asr0/WIP_asr0_deepspeech2_online_wenetspeech_ckpt_1.0.0a.model.tar.gz',
+        'https://paddlespeech.bj.bcebos.com/s2t/wenetspeech/asr0/asr0_deepspeech2_online_wenetspeech_ckpt_1.0.0a.model.tar.gz',
         'md5':
-        'b3ef6fcae8c0058c3c53375341ccb209',
+        'e393d4d274af0f6967db24fc146e8074',
         'cfg_path':
         'model.yaml',
         'ckpt_path':
-        'exp/deepspeech2_online/checkpoints/avg_3',
+        'exp/deepspeech2_online/checkpoints/avg_10',
         'lm_url':
         'https://deepspeech.bj.bcebos.com/zh_lm/zh_giga.no_cna_cmn.prune01244.klm',
         'lm_md5':
diff --git a/tests/unit/cli/test_cli.sh b/tests/unit/cli/test_cli.sh
index bdf05524..e1f1853f 100755
--- a/tests/unit/cli/test_cli.sh
+++ b/tests/unit/cli/test_cli.sh
@@ -14,6 +14,7 @@ wget -c https://paddlespeech.bj.bcebos.com/PaddleAudio/zh.wav https://paddlespee
 paddlespeech asr --input ./zh.wav
 paddlespeech asr --model conformer_aishell --input ./zh.wav
 paddlespeech asr --model conformer_online_aishell --input ./zh.wav
+paddlespeech asr --model conformer_online_wenetspeech --input ./zh.wav
 paddlespeech asr --model conformer_online_multicn --input ./zh.wav
 paddlespeech asr --model transformer_librispeech --lang en --input ./en.wav
 paddlespeech asr --model deepspeech2offline_aishell --input ./zh.wav