From 4f498e8d82ae5e7e977326a24f897c4b9fdc46bb Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期六, 06 五月 2023 22:24:50 +0800
Subject: [PATCH] websocket
---
funasr/runtime/python/websocket/README.md | 4 ++--
funasr/runtime/python/websocket/ws_client.py | 6 +++++-
2 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/funasr/runtime/python/websocket/README.md b/funasr/runtime/python/websocket/README.md
index ee7dca0..fd48f36 100644
--- a/funasr/runtime/python/websocket/README.md
+++ b/funasr/runtime/python/websocket/README.md
@@ -54,12 +54,12 @@
#### Recording from mircrophone
```shell
# --chunk_size, "5,10,5"=600ms, "8,8,4"=480ms
-python ws_client.py --host "127.0.0.1" --port 10095 --chunk_size "5,10,5"
+python ws_client.py --host "127.0.0.1" --port 10095 --chunk_size "5,10,5" --words_max_print 100
```
#### Loadding from wav.scp(kaldi style)
```shell
# --chunk_size, "5,10,5"=600ms, "8,8,4"=480ms
-python ws_client.py --host "127.0.0.1" --port 10095 --chunk_size "5,10,5" --audio_in "./data/wav.scp"
+python ws_client.py --host "127.0.0.1" --port 10095 --chunk_size "5,10,5" --audio_in "./data/wav.scp" --words_max_print 100
```
## Acknowledge
diff --git a/funasr/runtime/python/websocket/ws_client.py b/funasr/runtime/python/websocket/ws_client.py
index c62dd87..d32ce0a 100644
--- a/funasr/runtime/python/websocket/ws_client.py
+++ b/funasr/runtime/python/websocket/ws_client.py
@@ -36,6 +36,10 @@
type=int,
default=1,
help="test_thread_num")
+parser.add_argument("--words_max_print",
+ type=int,
+ default=100,
+ help="chunk")
args = parser.parse_args()
args.chunk_size = [int(x) for x in args.chunk_size.split(",")]
@@ -152,7 +156,7 @@
# print(meg, end = '')
# print("\r")
text_print += " {}".format(meg["text"][0])
- text_print = text_print[-55:]
+ text_print = text_print[-args.words_max_print:]
#os.system('clear')
print("\r"+str(id)+":"+text_print)
except Exception as e:
--
Gitblit v1.9.1