speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix problem about chunk_interval

Browse Source
This commit is contained in:
zhaomingwork 2023-05-06 13:53:28 +00:00
parent af346ddd4d
commit eb6c31aedd
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/runtime/python/websocket/ws_client.py
View File

@ -151,8 +151,7 @@ async def message(id):
meg = json.loads(meg)
# print(meg, end = '')
# print("\r")
text = meg["text"]
text_print += text
text_print += " {}".format(meg["text"][0])
text_print = text_print[-55:]
#os.system('clear')
print("\r"+str(id)+":"+text_print)

9
funasr/runtime/python/websocket/ws_server_online.py
View File

@ -54,7 +54,11 @@ async def ws_serve(websocket, path):
websocket.param_dict_asr_online["chunk_size"] = message["chunk_size"]
await async_asr_online(websocket,audio)
frames_online.append(audio)
if len(frames_online) % message["chunk_interval"] == 0 or not is_speaking:
audio_in = b"".join(frames_online)
await async_asr_online(websocket,audio_in)
frames_online = []
@ -75,7 +79,8 @@ async def async_asr_online(websocket,audio_in): # ASR推理
websocket.param_dict_asr_online["cache"] = dict()
if "text" in rec_result:
if rec_result["text"] != "sil" and rec_result["text"] != "waiting_for_more_voice":
print(rec_result["text"])
if len(rec_result["text"])>0:
rec_result["text"][0]=rec_result["text"][0].replace(" ","")
message = json.dumps({"mode": "online", "text": rec_result["text"]})
await websocket.send(message)