ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: asr

Browse Source
This commit is contained in:
0Xiao0
2024-10-11 20:45:06 +08:00
parent 416978f6c6
commit b17fcf0feb
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/blackbox/asr.py
View File

@ -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