Add Python API for punctuation models. (#762)

2024-04-13 13:28:17 +08:00
parent 329fe1aa8b
commit 68b8b88b5a
14 changed files with 136 additions and 6 deletions
--- a/sherpa-onnx/python/csrc/offline-punctuation.h
+++ b/sherpa-onnx/python/csrc/offline-punctuation.h
@@ -0,0 +1,16 @@
+// sherpa-onnx/python/csrc/offline-punctuation.h
+//
+// Copyright (c)  2024  Xiaomi Corporation
+
+#ifndef SHERPA_ONNX_PYTHON_CSRC_OFFLINE_PUNCTUATION_H_
+#define SHERPA_ONNX_PYTHON_CSRC_OFFLINE_PUNCTUATION_H_
+
+#include "sherpa-onnx/python/csrc/sherpa-onnx.h"
+
+namespace sherpa_onnx {
+
+void PybindOfflinePunctuation(py::module *m);
+
+}
+
+#endif  // SHERPA_ONNX_PYTHON_CSRC_OFFLINE_PUNCTUATION_H_