ZK-AI
/
livetalking
mirror of https://gitee.com/dumengxue/livetalking.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix tts audio decode

This commit is contained in:
lihengzhong 2024-01-08 14:44:38 +08:00
parent deac619d5b
commit 6a7af5d006
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
asrreal.py
View File

@ -383,21 +383,9 @@ class ASR:
def push_audio(self,buffer): def push_audio(self,buffer):
print(f'[INFO] push_audio {len(buffer)}') print(f'[INFO] push_audio {len(buffer)}')
if len(buffer)>0: # if len(buffer)>0:
byte_stream=BytesIO(buffer) # byte_stream=BytesIO(buffer)
stream = self.create_bytes_stream(byte_stream) # stream = self.create_bytes_stream(byte_stream)
streamlen = stream.shape[0]
idx=0
while streamlen >= self.chunk:
self.queue.put(stream[idx:idx+self.chunk])
streamlen -= self.chunk
idx += self.chunk
if streamlen>0:
self.queue.put(stream[idx:])
# self.input_stream.write(buffer)
# if len(buffer)<=0:
# self.input_stream.seek(0)
# stream = self.create_bytes_stream(self.input_stream)
# streamlen = stream.shape[0] # streamlen = stream.shape[0]
# idx=0 # idx=0
# while streamlen >= self.chunk: # while streamlen >= self.chunk:
@ -406,6 +394,18 @@ class ASR:
# idx += self.chunk # idx += self.chunk
# if streamlen>0: # if streamlen>0:
# self.queue.put(stream[idx:]) # self.queue.put(stream[idx:])
self.input_stream.write(buffer)
if len(buffer)<=0:
self.input_stream.seek(0)
stream = self.create_bytes_stream(self.input_stream)
streamlen = stream.shape[0]
idx=0
while streamlen >= self.chunk:
self.queue.put(stream[idx:idx+self.chunk])
streamlen -= self.chunk
idx += self.chunk
if streamlen>0:
self.queue.put(stream[idx:])
def get_audio_out(self): def get_audio_out(self):
return self.output_queue.get() return self.output_queue.get()