From f2a208d0d2e6a7805c44cd83faa7c313e01e594c Mon Sep 17 00:00:00 2001
From: SWHL <liekkaskono@163.com>
Date: 星期二, 21 二月 2023 15:38:38 +0800
Subject: [PATCH] Add code of generating whl about rapid_paraformer
---
funasr/runtime/python/onnxruntime/rapid_paraformer/paraformer_onnx.py | 19 ++++++++-----------
1 files changed, 8 insertions(+), 11 deletions(-)
diff --git a/funasr/runtime/python/onnxruntime/rapid_paraformer/paraformer_onnx.py b/funasr/runtime/python/onnxruntime/rapid_paraformer/paraformer_onnx.py
index e0c622c..a786ef0 100644
--- a/funasr/runtime/python/onnxruntime/rapid_paraformer/paraformer_onnx.py
+++ b/funasr/runtime/python/onnxruntime/rapid_paraformer/paraformer_onnx.py
@@ -2,28 +2,27 @@
# @Author: SWHL
# @Contact: liekkaskono@163.com
import os.path
-import traceback
from pathlib import Path
from typing import List, Union, Tuple
import librosa
import numpy as np
-from utils.utils import (CharTokenizer, Hypothesis, ONNXRuntimeError,
- OrtInferSession, TokenIDConverter, get_logger,
- read_yaml)
-from utils.postprocess_utils import sentence_postprocess
-from utils.frontend import WavFrontend
+from .utils.utils import (CharTokenizer, Hypothesis, ONNXRuntimeError,
+ OrtInferSession, TokenIDConverter, get_logger,
+ read_yaml)
+from .utils.postprocess_utils import sentence_postprocess
+from .utils.frontend import WavFrontend
logging = get_logger()
class Paraformer():
- def __init__(self, model_dir: Union[str, Path]=None,
+ def __init__(self, model_dir: Union[str, Path] = None,
batch_size: int = 1,
- device_id: Union[str, int]="-1",
+ device_id: Union[str, int] = "-1",
):
-
+
if not Path(model_dir).exists():
raise FileNotFoundError(f'{model_dir} does not exist.')
@@ -140,5 +139,3 @@
text = texts[0]
# text = self.tokenizer.tokens2text(token)
return text
-
-
--
Gitblit v1.9.1