From 0b5ab0709e8292447b314f4f02c74becafd6ce76 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 19 九月 2023 12:33:53 +0800
Subject: [PATCH] wechat
---
funasr/runtime/onnxruntime/include/offline-stream.h | 4 ++++
1 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/funasr/runtime/onnxruntime/include/offline-stream.h b/funasr/runtime/onnxruntime/include/offline-stream.h
index bc19bf2..21deb36 100644
--- a/funasr/runtime/onnxruntime/include/offline-stream.h
+++ b/funasr/runtime/onnxruntime/include/offline-stream.h
@@ -7,6 +7,7 @@
#include "model.h"
#include "punc-model.h"
#include "vad-model.h"
+#include "itn-model.h"
namespace funasr {
class OfflineStream {
@@ -17,12 +18,15 @@
std::unique_ptr<VadModel> vad_handle= nullptr;
std::unique_ptr<Model> asr_handle= nullptr;
std::unique_ptr<PuncModel> punc_handle= nullptr;
+ std::unique_ptr<ITNModel> itn_handle = nullptr;
bool UseVad(){return use_vad;};
bool UsePunc(){return use_punc;};
+ bool UseITN(){return use_itn;};
private:
bool use_vad=false;
bool use_punc=false;
+ bool use_itn=false;
};
OfflineStream *CreateOfflineStream(std::map<std::string, std::string>& model_path, int thread_num=1);
--
Gitblit v1.9.1