speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update model.h

Browse Source
This commit is contained in:
雾聪 2024-03-21 14:44:10 +08:00
parent d4aaa84ad1
commit 733930b2aa
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
runtime/onnxruntime/include/model.h
View File

@ -5,6 +5,10 @@
#include <string>
#include <map>
#include "funasrruntime.h"
#include "vocab.h"
#include "phone-set.h"
#include "fst/fstlib.h"
#include "fst/symbol-table.h"
namespace funasr {
class Model {
public:
@ -24,9 +28,9 @@ class Model {
virtual std::vector<std::vector<float>> CompileHotwordEmbedding(std::string &hotwords){return std::vector<std::vector<float>>();};
virtual std::string GetLang(){return "";};
virtual int GetAsrSampleRate() = 0;
virtual Vocab* GetVocab(){};
virtual Vocab* GetLmVocab(){};
virtual PhoneSet* GetPhoneSet(){};
virtual Vocab* GetVocab() {return nullptr;};
virtual Vocab* GetLmVocab() {return nullptr;};
virtual PhoneSet* GetPhoneSet() {return nullptr;};
std::shared_ptr<fst::Fst<fst::StdArc>> lm_ = nullptr;
};