Yabin Li
2024-06-25 b7060884fa4b8b85f79462644a5c99062d223da0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
#ifndef ITN_MODEL_H
#define ITN_MODEL_H
 
#include <string>
#include <map>
#include <vector>
 
namespace funasr {
class ITNModel {
  public:
    virtual ~ITNModel(){};
    virtual void InitITN(const std::string &itn_tagger, const std::string &itn_verbalizer, int thread_num)=0;
    virtual std::string Normalize(const std::string& input){return "";};
};
 
ITNModel *CreateITNModel(std::map<std::string, std::string>& model_path, int thread_num);
 
} // namespace funasr
#endif