Add online punctuation and casing prediction model for English language (#1224)

This commit is contained in:
jianyou
2024-08-06 17:33:38 +08:00
committed by GitHub
parent 52830cc910
commit 1414e4dc61
14 changed files with 874 additions and 0 deletions

View File

@@ -12,6 +12,7 @@
#include "sherpa-onnx/csrc/offline-lm-config.h"
#include "sherpa-onnx/csrc/offline-model-config.h"
#include "sherpa-onnx/csrc/offline-punctuation-model-config.h"
#include "sherpa-onnx/csrc/online-punctuation-model-config.h"
#include "sherpa-onnx/csrc/online-lm-config.h"
#include "sherpa-onnx/csrc/online-model-config.h"
#include "sherpa-onnx/csrc/speaker-embedding-extractor.h"
@@ -52,6 +53,9 @@ Ort::SessionOptions GetSessionOptions(const AudioTaggingModelConfig &config);
Ort::SessionOptions GetSessionOptions(
const OfflinePunctuationModelConfig &config);
Ort::SessionOptions GetSessionOptions(
const OnlinePunctuationModelConfig &config);
} // namespace sherpa_onnx
#endif // SHERPA_ONNX_CSRC_SESSION_H_