EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add Java and Kotlin API for punctuation models (#818)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-04-26 22:06:48 +08:00
committed by GitHub
parent db25986240
commit 5407f880c0
19 changed files with 515 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
sherpa-onnx/csrc/offline-punctuation-impl.cc
View File

@@ -4,6 +4,11 @@
#include "sherpa-onnx/csrc/offline-punctuation-impl.h"
#if __ANDROID_API__ >= 9
#include "android/asset_manager.h"
#include "android/asset_manager_jni.h"
#endif
#include "sherpa-onnx/csrc/macros.h"
#include "sherpa-onnx/csrc/offline-punctuation-ct-transformer-impl.h"
@@ -19,4 +24,16 @@ std::unique_ptr<OfflinePunctuationImpl> OfflinePunctuationImpl::Create(
return nullptr;
}
#if __ANDROID_API__ >= 9
std::unique_ptr<OfflinePunctuationImpl> OfflinePunctuationImpl::Create(
AAssetManager *mgr, const OfflinePunctuationConfig &config) {
if (!config.model.ct_transformer.empty()) {
return std::make_unique<OfflinePunctuationCtTransformerImpl>(mgr, config);
}
SHERPA_ONNX_LOGE("Please specify a punctuation model! Return a null pointer");
return nullptr;
}
#endif
} // namespace sherpa_onnx