funasr/runtime/onnxruntime/src/vocab.h
@@ -7,10 +7,10 @@ #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); @@ -18,8 +18,10 @@ Vocab(const char *filename); ~Vocab(); int Size(); string Vector2String(vector<int> in); bool IsChinese(string ch); void Vector2String(vector<int> in, std::vector<std::string> &preds); string Vector2StringV2(vector<int> in); }; } // namespace funasr #endif