From f5825f1f4f228e459455e7b739bf3b7e2aac3395 Mon Sep 17 00:00:00 2001
From: mengzhe.cmz <mengzhe.cmz@alibaba-inc.com>
Date: 星期一, 08 五月 2023 18:54:30 +0800
Subject: [PATCH] onnx runtime model int32
---
funasr/runtime/python/websocket/ws_server_offline.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/funasr/runtime/python/websocket/ws_server_offline.py b/funasr/runtime/python/websocket/ws_server_offline.py
index c2366d0..15578f6 100644
--- a/funasr/runtime/python/websocket/ws_server_offline.py
+++ b/funasr/runtime/python/websocket/ws_server_offline.py
@@ -96,6 +96,7 @@
if not is_speaking:
websocket.vad_pre_idx = 0
frames = []
+ websocket.param_dict_vad = {'in_cache': dict()}
else:
frames = frames[-20:]
@@ -137,10 +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"]], "wav_name": websocket.wav_name})
+ message = json.dumps({"mode": "offline", "text": rec_result["text"], "wav_name": websocket.wav_name})
await websocket.send(message)
-
+
--
Gitblit v1.9.1