fix pretrain model error

paddlespeech/resource/pretrained_models.py

@@ -175,7 +175,7 @@ asr_dynamic_pretrained_models = {
             'ckpt_path':'exp/deepspeech2_online/checkpoints/avg_1',
             'model':'exp/deepspeech2_online/checkpoints/avg_1.jit.pdmodel',
             'params':'exp/deepspeech2_online/checkpoints/avg_1.jit.pdiparams',
-            'onnx_model': 'onnx/model.onnx'
+            'onnx_model': 'onnx/model.onnx',
             'lm_url':'https://deepspeech.bj.bcebos.com/zh_lm/zh_giga.no_cna_cmn.prune01244.klm',
             'lm_md5':'29e02312deb2e59b3c8686c7966d4fe3'
         },
@@ -228,7 +228,7 @@ asr_static_pretrained_models = {
             'ckpt_path':'exp/deepspeech2_online/checkpoints/avg_1',
             'model':'exp/deepspeech2_online/checkpoints/avg_1.jit.pdmodel',
             'params':'exp/deepspeech2_online/checkpoints/avg_1.jit.pdiparams',
-            'onnx_model': 'onnx/model.onnx'
+            'onnx_model': 'onnx/model.onnx',
             'lm_url':'https://deepspeech.bj.bcebos.com/zh_lm/zh_giga.no_cna_cmn.prune01244.klm',
             'lm_md5':'29e02312deb2e59b3c8686c7966d4fe3'
         },
@@ -246,7 +246,7 @@ asr_onnx_pretrained_models = {
             'ckpt_path':'exp/deepspeech2_online/checkpoints/avg_1',
             'model':'exp/deepspeech2_online/checkpoints/avg_1.jit.pdmodel',
             'params':'exp/deepspeech2_online/checkpoints/avg_1.jit.pdiparams',
-            'onnx_model': 'onnx/model.onnx'
+            'onnx_model': 'onnx/model.onnx',
             'lm_url':'https://deepspeech.bj.bcebos.com/zh_lm/zh_giga.no_cna_cmn.prune01244.klm',
             'lm_md5':'29e02312deb2e59b3c8686c7966d4fe3'
         },

speechx/examples/ds2_ol/onnx/local/infer_check.py

@@ -27,7 +27,8 @@ def parse_args():
         '--input_file',
         type=str,
         default="static_ds2online_inputs.pickle",
-        help="aishell ds2 input data file. For wenetspeech, we only feed for infer model", )
+        help="aishell ds2 input data file. For wenetspeech, we only feed for infer model",
+    )
     parser.add_argument(
         '--model_type',
         type=str,
@@ -57,7 +58,6 @@ if __name__ == '__main__':
         iodict = pickle.load(f)
         print(iodict.keys())
 
-
     audio_chunk = iodict['audio_chunk']
     audio_chunk_lens = iodict['audio_chunk_lens']
     chunk_state_h_box = iodict['chunk_state_h_box']