Browse Source

add fire engine manually

zhaomingwork 2 năm trước cách đây
mục cha
commit
c00336e115

+ 11 - 1
funasr/runtime/python/websocket/parse_args.py

@@ -35,6 +35,16 @@ parser.add_argument("--ncpu",
                     type=int,
                     default=1,
                     help="cpu cores")
-
+parser.add_argument("--certfile",
+                    type=str,
+                    default="",
+                    required=False,
+                    help="certfile for ssl")
+                    
+parser.add_argument("--keyfile",
+                    type=str,
+                    default="",
+                    required=False,
+                    help="keyfile for ssl")
 args = parser.parse_args()
 print(args)

+ 4 - 1
funasr/runtime/python/websocket/ws_server_online.py

@@ -53,6 +53,9 @@ async def ws_serve(websocket, path):
 				if "is_speaking" in messagejson:
 					websocket.is_speaking = messagejson["is_speaking"]
 					websocket.param_dict_asr_online["is_final"] = not websocket.is_speaking
+					# need to fire engine manually if no data received any more
+					if not websocket.is_speaking:
+						await async_asr_online(websocket,b"")
 				if "chunk_interval" in messagejson:
 					websocket.chunk_interval=messagejson["chunk_interval"]
 				if "wav_name" in messagejson:
@@ -82,7 +85,7 @@ async def ws_serve(websocket, path):
 
 
 async def async_asr_online(websocket,audio_in):
-	if len(audio_in) > 0:
+	if len(audio_in) >=0:
 		audio_in = load_bytes(audio_in)
 		rec_result = inference_pipeline_asr_online(audio_in=audio_in,
 		                                           param_dict=websocket.param_dict_asr_online)