From 4853dffdc4e70552f5b4348ef29089ed04927d95 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 17 二月 2023 11:43:40 +0800
Subject: [PATCH] Merge pull request #131 from alibaba-damo-academy/dev_lyb
---
funasr/runtime/python/onnxruntime/paraformer/rapid_paraformer/paraformer_onnx.py | 9 +++++----
1 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/funasr/runtime/python/onnxruntime/paraformer/rapid_paraformer/paraformer_onnx.py b/funasr/runtime/python/onnxruntime/paraformer/rapid_paraformer/paraformer_onnx.py
index 621b903..64dbaf8 100644
--- a/funasr/runtime/python/onnxruntime/paraformer/rapid_paraformer/paraformer_onnx.py
+++ b/funasr/runtime/python/onnxruntime/paraformer/rapid_paraformer/paraformer_onnx.py
@@ -41,8 +41,8 @@
self.ort_infer = OrtInferSession(model_file, device_id)
self.batch_size = batch_size
- def __call__(self, wav_content: Union[str, np.ndarray, List[str]], fs: int = None) -> List:
- waveform_list = self.load_data(wav_content, fs)
+ def __call__(self, wav_content: Union[str, np.ndarray, List[str]], **kwargs) -> List:
+ waveform_list = self.load_data(wav_content, self.frontend.opts.samp_freq)
waveform_nums = len(waveform_list)
asr_res = []
@@ -54,8 +54,9 @@
try:
am_scores, valid_token_lens = self.infer(feats, feats_len)
except ONNXRuntimeError:
- logging.error(traceback.format_exc())
- preds = []
+ #logging.warning(traceback.format_exc())
+ logging.warning("input wav is silence or noise")
+ preds = ['']
else:
preds = self.decode(am_scores, valid_token_lens)
--
Gitblit v1.9.1