From 7ab2e5cf22bbb31808bcacf84c054c710e4e6a93 Mon Sep 17 00:00:00 2001
From: Yabin Li <wucong.lyb@alibaba-inc.com>
Date: 星期一, 24 四月 2023 16:19:17 +0800
Subject: [PATCH] Merge pull request #400 from alibaba-damo-academy/dev_knf
---
funasr/runtime/onnxruntime/src/vocab.h | 25 +++++++++++++++++++++++++
1 files changed, 25 insertions(+), 0 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/vocab.h b/funasr/runtime/onnxruntime/src/vocab.h
new file mode 100644
index 0000000..a3fdf65
--- /dev/null
+++ b/funasr/runtime/onnxruntime/src/vocab.h
@@ -0,0 +1,25 @@
+
+#ifndef VOCAB_H
+#define VOCAB_H
+
+#include <stdint.h>
+#include <string>
+#include <vector>
+using namespace std;
+
+class Vocab {
+ private:
+ vector<string> vocab;
+ 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);
+};
+
+#endif
--
Gitblit v1.9.1