diff --git a/paddlespeech/cli/asr/infer.py b/paddlespeech/cli/asr/infer.py index 050d6f7f9c962a4f0b870e1cab214eb2397d41cf..0fb5486837d1cf8795850b69fa58ba0a8e000a00 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") - exit(1) + 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 c00e5e2940e67fd10ad2ae94307ceb9e2694328f..926b1ac0ba04452c42b2057f521c5c8952fe8086 100755 --- a/tests/unit/cli/test_cli.sh +++ b/tests/unit/cli/test_cli.sh @@ -22,7 +22,7 @@ paddlespeech asr --model deepspeech2offline_librispeech --lang en --input ./en.w # 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