diff --git a/paddlespeech/cli/asr/infer.py b/paddlespeech/cli/asr/infer.py index 97a1b3219..0fb548683 100644 --- a/paddlespeech/cli/asr/infer.py +++ b/paddlespeech/cli/asr/infer.py @@ -346,7 +346,7 @@ class ASRExecutor(BaseExecutor): max_duration = 50.0 if audio_duration >= max_duration: logger.error("Please input audio file less then 50 seconds.\n") - return + return False except Exception as e: logger.exception(e) logger.error( @@ -383,7 +383,7 @@ class ASRExecutor(BaseExecutor): ) == "n" or content.strip() == "no" or content.strip( ) == "No": logger.info("Exit the program") - exit(1) + return False else: logger.warning("Not regular input, please input again") diff --git a/tests/unit/cli/test_cli.sh b/tests/unit/cli/test_cli.sh index 96e1e0235..926b1ac0b 100755 --- a/tests/unit/cli/test_cli.sh +++ b/tests/unit/cli/test_cli.sh @@ -12,12 +12,17 @@ paddlespeech text --input 今天的天气真不错啊你下午有空吗我想约 # Speech_recognition wget -c https://paddlespeech.bj.bcebos.com/PaddleAudio/zh.wav https://paddlespeech.bj.bcebos.com/PaddleAudio/en.wav 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 transformer_librispeech --lang en --input ./en.wav +paddlespeech asr --model deepspeech2offline_aishell --input ./zh.wav +paddlespeech asr --model deepspeech2online_aishell --input ./zh.wav +paddlespeech asr --model deepspeech2offline_librispeech --lang en --input ./en.wav # long audio restriction wget -c wget https://paddlespeech.bj.bcebos.com/datasets/single_wav/zh/test_long_audio_01.wav paddlespeech asr --input test_long_audio_01.wav -if [ $? -ne 1 ]; then +if [ $? -ne -1 ]; then exit 1 fi