1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
| #ifndef MODEL_H
| #define MODEL_H
|
| #include <string>
| #include <map>
| #include "funasrruntime.h"
| namespace funasr {
| class Model {
| public:
| virtual ~Model(){};
| virtual void Reset() = 0;
| virtual void InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, int thread_num){};
| virtual void InitAsr(const std::string &en_model, const std::string &de_model, const std::string &am_cmvn, const std::string &am_config, int thread_num){};
| virtual void InitAsr(const std::string &am_model, const std::string &en_model, const std::string &de_model, const std::string &am_cmvn, const std::string &am_config, int thread_num){};
| virtual std::string Forward(float *din, int len, bool input_finished, const std::vector<std::vector<float>> &hw_emb={{0.0}}){return "";};
| virtual std::string Rescoring() = 0;
| virtual void InitHwCompiler(const std::string &hw_model, int thread_num){};
| virtual void InitSegDict(const std::string &seg_dict_model){};
| virtual std::vector<std::vector<float>> CompileHotwordEmbedding(std::string &hotwords){};
| };
|
| Model *CreateModel(std::map<std::string, std::string>& model_path, int thread_num=1, ASR_TYPE type=ASR_OFFLINE);
| Model *CreateModel(void* asr_handle, std::vector<int> chunk_size);
|
| } // namespace funasr
| #endif
|
|
