diff --git a/funasr/runtime/python/websocket/README.md b/funasr/runtime/python/websocket/README.md index ee7dca0a9..fd48f3609 100644 --- a/funasr/runtime/python/websocket/README.md +++ b/funasr/runtime/python/websocket/README.md @@ -54,12 +54,12 @@ pip install -r requirements_client.txt #### 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 c62dd8706..d32ce0a5a 100644 --- a/funasr/runtime/python/websocket/ws_client.py +++ b/funasr/runtime/python/websocket/ws_client.py @@ -36,6 +36,10 @@ parser.add_argument("--test_thread_num", 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 @@ async def message(id): # 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: