diff --git a/ov_infer.py b/ov_infer.py index df37689..967706a 100644 --- a/ov_infer.py +++ b/ov_infer.py @@ -12,7 +12,7 @@ import numpy as np from pathlib import Path from typing import List, Union, Tuple -import openvino_tokenizers +import openvino_tokenizers # Required to register SentencepieceDetokenizer operation for OpenVINO from funasr_onnx.utils.utils import ( # CharTokenizer, @@ -244,7 +244,7 @@ def infer( language: np.ndarray, textnorm: np.ndarray, ) -> Tuple[np.ndarray, np.ndarray]: - self.infer_reqeust.infer({"speech":feats, "speech_lengths":feats_len, "language":language,"textnorm":textnorm}) + self.infer_reqeust.infer([feats, feats_len, language, textnorm]) ctc_logits = self.infer_reqeust.get_tensor('ctc_logits').data.copy() encoder_out_lens = self.infer_reqeust.get_tensor('encoder_out_lens').data.copy() return ctc_logits, encoder_out_lens