author:yijinsheng wss_client_fix

modelscope · Dec 15, 2023 · 3c2d793 · 3c2d793
1 parent b0b796f
commit 3c2d793
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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)