speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

onnx runtime

Browse Source
This commit is contained in:
游雁 2023-02-16 20:49:23 +08:00
parent b841635a53
commit 41e91414fa
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/runtime/python/onnxruntime/paraformer/rapid_paraformer/paraformer_onnx.py
View File

@ -41,8 +41,8 @@ class Paraformer():
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 = []