1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
| #ifndef MODEL_H
| #define MODEL_H
|
| #include <string>
|
| class Model {
| public:
| virtual ~Model(){};
| virtual void reset() = 0;
| virtual std::string forward_chunk(float *din, int len, int flag) = 0;
| virtual std::string forward(float *din, int len, int flag) = 0;
| virtual std::string rescoring() = 0;
| virtual std::vector<std::vector<int>> vad_seg(std::vector<float>& pcm_data)=0;
| virtual std::string AddPunc(const char* szInput)=0;
| };
|
| Model *CreateModel(const char *path,int nThread=0,bool quantize=false, bool use_vad=false, bool use_punc=false);
| #endif
|
|
