志浩
2023-08-01 5cfdcfc45a042e338c2b2f4a08dab125de3fb5ee
funasr/runtime/onnxruntime/src/vocab.h
@@ -7,19 +7,21 @@
#include <vector>
using namespace std;
namespace funasr {
class Vocab {
  private:
    vector<string> vocab;
    bool isChinese(string ch);
    bool isEnglish(string ch);
    void loadVocabFromYaml(const char* filename);
    bool IsChinese(string ch);
    bool IsEnglish(string ch);
    void LoadVocabFromYaml(const char* filename);
  public:
    Vocab(const char *filename);
    ~Vocab();
    int size();
    string vector2string(vector<int> in);
    string vector2stringV2(vector<int> in);
    int Size();
    string Vector2String(vector<int> in);
    string Vector2StringV2(vector<int> in);
};
} // namespace funasr
#endif