From 4413d1eb47fa400277c8a9625aa0bd5c424a2fab Mon Sep 17 00:00:00 2001
From: zhaomingwork <61895407+zhaomingwork@users.noreply.github.com>
Date: 星期四, 06 七月 2023 10:13:32 +0800
Subject: [PATCH] Python ws client slow problem for multiple files in offline mode (#716)

---
 funasr/runtime/python/websocket/funasr_wss_client.py |    7 +++----
 1 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/funasr/runtime/python/websocket/funasr_wss_client.py b/funasr/runtime/python/websocket/funasr_wss_client.py
index fad12ce..765addb 100644
--- a/funasr/runtime/python/websocket/funasr_wss_client.py
+++ b/funasr/runtime/python/websocket/funasr_wss_client.py
@@ -177,10 +177,9 @@
             sleep_duration = 0.001 if args.mode == "offline" else 60 * args.chunk_size[1] / args.chunk_interval / 1000
             
             await asyncio.sleep(sleep_duration)
-    # when all data sent, we need to close websocket
-    while not voices.empty():
-         await asyncio.sleep(1)
-    await asyncio.sleep(3)
+    
+    if not args.mode=="offline":
+        await asyncio.sleep(2)
     # offline model need to wait for message recved
     
     if args.mode=="offline":

--
Gitblit v1.9.1