From 3c2d793370d11c32fede02148478899be1a5a501 Mon Sep 17 00:00:00 2001 From: lyblsgo Date: Fri, 15 Dec 2023 15:35:53 +0800 Subject: [PATCH] author:yijinsheng wss_client_fix --- runtime/python/websocket/funasr_wss_client.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/runtime/python/websocket/funasr_wss_client.py b/runtime/python/websocket/funasr_wss_client.py index bfe4b0d7a..e469319c9 100644 --- a/runtime/python/websocket/funasr_wss_client.py +++ b/runtime/python/websocket/funasr_wss_client.py @@ -256,6 +256,7 @@ async def message(id): wav_name = meg.get("wav_name", "demo") text = meg["text"] timestamp="" + offline_msg_done = meg.get("is_final", False) if "timestamp" in meg: timestamp = meg["timestamp"] @@ -265,7 +266,9 @@ async def message(id): else: text_write_line = "{}\t{}\n".format(wav_name, text) ibest_writer.write(text_write_line) - + + if 'mode' not in meg: + continue if meg["mode"] == "online": text_print += "{}".format(text) text_print = text_print[-args.words_max_print:] @@ -292,7 +295,7 @@ async def message(id): text_print = text_print[-args.words_max_print:] os.system('clear') print("\rpid" + str(id) + ": " + text_print) - offline_msg_done=True + # offline_msg_done=True except Exception as e: print("Exception:", e)