From 580b11b57ac4b62f7e2acda73813a4e10e8e4cd3 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 10 十月 2023 17:17:29 +0800
Subject: [PATCH] v0.8.0

---
 funasr/runtime/onnxruntime/src/paraformer.h |    4 ++++
 1 files changed, 4 insertions(+), 0 deletions(-)

diff --git a/funasr/runtime/onnxruntime/src/paraformer.h b/funasr/runtime/onnxruntime/src/paraformer.h
index 4080881..455078e 100644
--- a/funasr/runtime/onnxruntime/src/paraformer.h
+++ b/funasr/runtime/onnxruntime/src/paraformer.h
@@ -20,6 +20,7 @@
         //const float scale = 22.6274169979695;
         const float scale = 1.0;
 
+        void LoadConfigFromYaml(const char* filename);
         void LoadOnlineConfigFromYaml(const char* filename);
         void LoadCmvn(const char *filename);
         vector<float> ApplyLfr(const vector<float> &in);
@@ -53,6 +54,7 @@
         string PostProcess(std::vector<string> &raw_char, std::vector<std::vector<float>> &timestamp_list);
 
         string Rescoring();
+        string GetLang(){return language;};
 
         knf::FbankOptions fbank_opts_;
         vector<float> means_list_;
@@ -69,6 +71,8 @@
         vector<const char*> m_szInputNames;
         vector<const char*> m_szOutputNames;
 
+        std::string language="zh-cn";
+
         // paraformer-online
         std::shared_ptr<Ort::Session> encoder_session_ = nullptr;
         std::shared_ptr<Ort::Session> decoder_session_ = nullptr;

--
Gitblit v1.9.1