yumoqing/asr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2024-08-29 17:34:12 +08:00
parent 7e30812a09
commit 08369052c5
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
app/asr_engine.py
View File

@ -26,14 +26,16 @@ def save_base64_wav(base64_data, output_file,sample_rate=16000, num_channels=1):
async def generate(request, kw):
params_kw = kw.get('params_kw', DictObject())
model = params_kw.model
audio_file = params_kw.audio_file
if not audio_file:
audio = params_kw.audio
if audio is None:
return {
'status':'error',
'message':'audio is null'
}
fname = temp_file(suffix='.wav')
save_base64_wav(audio, fname)
audio_file = temp_file(suffix='.wav')
save_base64_wav(audio, audio_file)
engine = None
g = ServerEnv()
if model=='whisper':
@ -46,9 +48,9 @@ async def generate(request, kw):
}
try:
t1 = time.time()
dic = await engine.stt(fname)
dic = await engine.stt(audio_file)
t2 = time.time()
os.remove(fname)
os.remove(audio_file)
ret = {
"status":"ok",
"time_cost":t2-t1,