Chong Zhang
2023-05-23 5fec3c9e58fceda85fa2daf7deec2492372dac8a
funasr/runtime/python/websocket/ws_client.py
@@ -47,7 +47,7 @@
                    help="test_thread_num")
parser.add_argument("--words_max_print",
                    type=int,
                    default=100,
                    default=10000,
                    help="chunk")
parser.add_argument("--output_dir",
                    type=str,
@@ -181,7 +181,7 @@
                ibest_writer["text"][wav_name] = text
            
            if meg["mode"] == "online":
                text_print += " {}".format(text)
                text_print += "{}".format(text)
                text_print = text_print[-args.words_max_print:]
                os.system('clear')
                print("\rpid"+str(id)+": "+text_print)
@@ -192,7 +192,7 @@
                print("\rpid"+str(id)+": "+text_print)
            else:
                if meg["mode"] == "2pass-online":
                    text_print_2pass_online += " {}".format(text)
                    text_print_2pass_online += "{}".format(text)
                    text_print = text_print_2pass_offline + text_print_2pass_online
                else:
                    text_print_2pass_online = ""
@@ -255,7 +255,8 @@
          chunk_size=int((total_len)/args.test_thread_num)
          remain_wavs=total_len-chunk_size*args.test_thread_num
     else:
          chunk_size=0
          chunk_size=1
          remain_wavs=0
     process_list = []
     chunk_begin=0