From b17fcf0feb5227c1ad9255952e62f5ad38c92494 Mon Sep 17 00:00:00 2001 From: 0Xiao0 <511201264@qq.com> Date: Fri, 11 Oct 2024 20:45:06 +0800 Subject: [PATCH] fix: asr --- src/blackbox/asr.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/blackbox/asr.py b/src/blackbox/asr.py index cab6ea1..935f2f0 100644 --- a/src/blackbox/asr.py +++ b/src/blackbox/asr.py @@ -85,7 +85,7 @@ class ASR(Blackbox): print(f"asr_model_name: {user_model_name}") data = args[0] - if user_model_name == 'sensevoice' or ['sensevoice']: + if 'sensevoice' in user_model_name: with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_audio_file: temp_audio_file.write(data) temp_audio_path = temp_audio_file.name @@ -106,8 +106,7 @@ class ASR(Blackbox): results = converter.convert(results) return results - elif user_model_name == 'funasr' or ['funasr']: - + elif 'funasr' in user_model_name: results = converter.convert(results) return results