From 8cd8e25f180b57c74f1ae7fd39001e5af4c20f8b Mon Sep 17 00:00:00 2001
From: sugarcase <shi.pengteng@163.com>
Date: 星期三, 25 九月 2024 23:24:58 +0800
Subject: [PATCH] add demo.py for all kws models (#2107)
---
runtime/onnxruntime/src/phone-set.h | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/runtime/onnxruntime/src/phone-set.h b/runtime/onnxruntime/src/phone-set.h
index 9729104..cb1a2a7 100644
--- a/runtime/onnxruntime/src/phone-set.h
+++ b/runtime/onnxruntime/src/phone-set.h
@@ -5,6 +5,7 @@
#include <string>
#include <vector>
#include <unordered_map>
+#include "nlohmann/json.hpp"
#define UNIT_BEG_SIL_SYMBOL "<s>"
#define UNIT_END_SIL_SYMBOL "</s>"
#define UNIT_BLK_SYMBOL "<blank>"
@@ -28,6 +29,7 @@
vector<string> phone_;
unordered_map<string, int> phn2Id_;
void LoadPhoneSetFromYaml(const char* filename);
+ void LoadPhoneSetFromJson(const char* filename);
};
} // namespace funasr
--
Gitblit v1.9.1