mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
20 lines
579 B
C++
20 lines
579 B
C++
|
|
#ifndef MODEL_H
|
|
#define MODEL_H
|
|
|
|
#include <string>
|
|
|
|
class Model {
|
|
public:
|
|
virtual ~Model(){};
|
|
virtual void Reset() = 0;
|
|
virtual std::string ForwardChunk(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>> VadSeg(std::vector<float>& pcm_data)=0;
|
|
virtual std::string AddPunc(const char* sz_input)=0;
|
|
};
|
|
|
|
Model *CreateModel(const char *path,int thread_num=1,bool quantize=false, bool use_vad=false, bool use_punc=false);
|
|
#endif
|