Support getting word IDs for CTC HLG decoding. (#978)

2024-06-06 14:22:39 +08:00
parent 69347ffc8f
commit 1a43d1e37f
13 changed files with 60 additions and 13 deletions
--- a/sherpa-onnx/python/csrc/offline-stream.cc
+++ b/sherpa-onnx/python/csrc/offline-stream.cc
@@ -34,6 +34,8 @@ static void PybindOfflineRecognitionResult(py::module *m) {  // NOLINT
          })
      .def_property_readonly("tokens",
                             [](const PyClass &self) { return self.tokens; })
+      .def_property_readonly("words",
+                             [](const PyClass &self) { return self.words; })
      .def_property_readonly(
          "timestamps", [](const PyClass &self) { return self.timestamps; });
 }