fix readme demo run error.

1 year ago · c9a0b39f62
parent 7c73010c29
commit c9a0b39f62
1 changed files with 1 additions and 1 deletions
--- a/paddlespeech/cli/asr/infer.py
+++ b/paddlespeech/cli/asr/infer.py
@ -274,7 +274,7 @@ class ASRExecutor(BaseExecutor):
            # fbank
            audio = preprocessing(audio, **preprocess_args)

-            audio_len = paddle.to_tensor([audio.shape[0]]).unsqueeze(axis=0)
+            audio_len = paddle.to_tensor(audio.shape[0]).unsqueeze(axis=0)
            audio = paddle.to_tensor(audio, dtype='float32').unsqueeze(axis=0)

            self._inputs["audio"] = audio