Merge pull request #473 from alibaba-damo-academy/dev_smohan

jmwang66

2023-05-09 8dab6d184a034ca86eafa644ea0d2100aadfe27d

 funasr/runtime/python/websocket/ws_server_offline.py

@@ -78,6 +78,7 @@

                is_speaking = message["is_speaking"]
                websocket.param_dict_vad["is_final"] = not is_speaking
                websocket.wav_name = message.get("wav_name", "demo")
                if speech_start:
                    frames_asr.append(audio)
                speech_start_i, speech_end_i = await async_vad(websocket, audio)
@@ -95,8 +96,9 @@
                    if not is_speaking:
                        websocket.vad_pre_idx = 0
                        frames = []
                        websocket.param_dict_vad = {'in_cache': dict()}
                    else:
                        frames = frames[-10:]
                        frames = frames[-20:]

     
    except websockets.ConnectionClosed:
@@ -136,9 +138,10 @@
                    rec_result = inference_pipeline_punc(text_in=rec_result['text'],
                                                         param_dict=websocket.param_dict_punc)
                    # print(rec_result)
                    message = json.dumps({"mode": "offline", "text": [rec_result["text"]]})
                    await websocket.send(message)
                    
                message = json.dumps({"mode": "offline", "text": rec_result["text"], "wav_name": websocket.wav_name})
                await websocket.send(message)