From 676957a1e7000a948bb2cc876ecec0f1b7768afa Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 21 二月 2023 11:23:21 +0800
Subject: [PATCH] Merge pull request #140 from alibaba-damo-academy/dev_lzr
---
funasr/runtime/python/grpc/grpc_main_server.py | 16 ++++++++++++++--
1 files changed, 14 insertions(+), 2 deletions(-)
diff --git a/funasr/runtime/python/grpc/grpc_main_server.py b/funasr/runtime/python/grpc/grpc_main_server.py
index f3b2348..e862ac4 100644
--- a/funasr/runtime/python/grpc/grpc_main_server.py
+++ b/funasr/runtime/python/grpc/grpc_main_server.py
@@ -9,7 +9,8 @@
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10),
# interceptors=(AuthInterceptor('Bearer mysecrettoken'),)
)
- paraformer_pb2_grpc.add_ASRServicer_to_server(ASRServicer(args.user_allowed, args.model, args.sample_rate), server)
+ paraformer_pb2_grpc.add_ASRServicer_to_server(
+ ASRServicer(args.user_allowed, args.model, args.sample_rate, args.backend, args.onnx_dir), server)
port = "[::]:" + str(args.port)
server.add_insecure_port(port)
server.start()
@@ -37,7 +38,18 @@
parser.add_argument("--sample_rate",
type=int,
default=16000,
- help="audio sample_rate from client")
+ help="audio sample_rate from client")
+
+ parser.add_argument("--backend",
+ type=str,
+ default="pipeline",
+ choices=("pipeline", "onnxruntime"),
+ help="backend, optional modelscope pipeline or onnxruntime")
+
+ parser.add_argument("--onnx_dir",
+ type=str,
+ default="/nfs/models/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
+ help="onnx model dir")
--
Gitblit v1.9.1