From 7bdcefb198484d067419fde3eb02ab38b6d7884c Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 30 一月 2023 18:25:30 +0800
Subject: [PATCH] Merge pull request #48 from dyyzhmm/main
---
funasr/runtime/python/grpc/grpc_main_server.py | 46 ++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 46 insertions(+), 0 deletions(-)
diff --git a/funasr/runtime/python/grpc/grpc_main_server.py b/funasr/runtime/python/grpc/grpc_main_server.py
new file mode 100644
index 0000000..f3b2348
--- /dev/null
+++ b/funasr/runtime/python/grpc/grpc_main_server.py
@@ -0,0 +1,46 @@
+import grpc
+from concurrent import futures
+import argparse
+
+import paraformer_pb2_grpc
+from grpc_server import ASRServicer
+
+def serve(args):
+ 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)
+ port = "[::]:" + str(args.port)
+ server.add_insecure_port(port)
+ server.start()
+ print("grpc server started!")
+ server.wait_for_termination()
+
+if __name__ == '__main__':
+ parser = argparse.ArgumentParser()
+ parser.add_argument("--port",
+ type=int,
+ default=10095,
+ required=True,
+ help="grpc server port")
+
+ parser.add_argument("--user_allowed",
+ type=str,
+ default="project1_user1|project1_user2|project2_user3",
+ help="allowed user for grpc client")
+
+ parser.add_argument("--model",
+ type=str,
+ default="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch",
+ help="model from modelscope")
+
+ parser.add_argument("--sample_rate",
+ type=int,
+ default=16000,
+ help="audio sample_rate from client")
+
+
+
+ args = parser.parse_args()
+
+ serve(args)
--
Gitblit v1.9.1