From 4ba1011b42e041ee1d71448eefd7ef2e7bd61bb6 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 31 三月 2023 15:31:26 +0800
Subject: [PATCH] export

---
 funasr/runtime/grpc/paraformer_server.h |   17 ++++++++++++-----
 1 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/funasr/runtime/grpc/paraformer_server.h b/funasr/runtime/grpc/paraformer_server.h
index 614dd04..e42e041 100644
--- a/funasr/runtime/grpc/paraformer_server.h
+++ b/funasr/runtime/grpc/paraformer_server.h
@@ -4,9 +4,6 @@
 #include <iostream>
 #include <memory>
 #include <string>
-#include <unordered_map>
-#include <chrono>
-#include <sstream>
 
 #include <grpc/grpc.h>
 #include <grpcpp/server.h>
@@ -14,8 +11,11 @@
 #include <grpcpp/server_context.h>
 #include <grpcpp/security/server_credentials.h>
 
-#include "paraformer.grpc.pb.h"
+#include <unordered_map>
+#include <chrono>
 
+#include "paraformer.grpc.pb.h"
+#include "librapidasrapi.h"
 
 
 using grpc::Server;
@@ -31,6 +31,12 @@
 using paraformer::Response;
 using paraformer::ASR;
 
+typedef struct
+{
+    std::string msg;
+    float  snippet_time;
+}RPASR_RECOG_RESULT;
+
 
 class ASRServicer final : public ASR::Service {
   private:
@@ -39,11 +45,12 @@
     std::unordered_map<std::string, std::string> client_transcription;
 
   public:
-    ASRServicer();
+    ASRServicer(const char* model_path, int thread_num, bool quantize);
     void clear_states(const std::string& user);
     void clear_buffers(const std::string& user);
     void clear_transcriptions(const std::string& user);
     void disconnect(const std::string& user);
     grpc::Status Recognize(grpc::ServerContext* context, grpc::ServerReaderWriter<Response, Request>* stream);
+    RPASR_HANDLE AsrHanlde;
 	
 };

--
Gitblit v1.9.1