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

Swift API for hotwords support (#331)

Browse Source
This commit is contained in:
zr_jin
2023-09-21 20:32:13 +08:00
committed by GitHub
parent 532ed142d2
commit b640c295b9
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
swift-api-examples/SherpaOnnx.swift
View File

@@ -108,7 +108,9 @@ func sherpaOnnxOnlineRecognizerConfig(
rule2MinTrailingSilence: Float = 1.2, rule2MinTrailingSilence: Float = 1.2,
rule3MinUtteranceLength: Float = 30, rule3MinUtteranceLength: Float = 30,
decodingMethod: String = "greedy_search", decodingMethod: String = "greedy_search",
maxActivePaths: Int = 4 maxActivePaths: Int = 4,
hotwordsFile: String = "",
hotwordsScore: Float = 1.5
) -> SherpaOnnxOnlineRecognizerConfig { ) -> SherpaOnnxOnlineRecognizerConfig {
return SherpaOnnxOnlineRecognizerConfig( return SherpaOnnxOnlineRecognizerConfig(
feat_config: featConfig, feat_config: featConfig,
@@ -118,7 +120,9 @@ func sherpaOnnxOnlineRecognizerConfig(
enable_endpoint: enableEndpoint ? 1 : 0, enable_endpoint: enableEndpoint ? 1 : 0,
rule1_min_trailing_silence: rule1MinTrailingSilence, rule1_min_trailing_silence: rule1MinTrailingSilence,
rule2_min_trailing_silence: rule2MinTrailingSilence, rule2_min_trailing_silence: rule2MinTrailingSilence,
rule3_min_utterance_length: rule3MinUtteranceLength) rule3_min_utterance_length: rule3MinUtteranceLength,
hotwords_file: toCPointer(hotwordsFile),
hotwords_score: hotwordsScore)
} }
/// Wrapper for recognition result. /// Wrapper for recognition result.
@@ -327,14 +331,18 @@ func sherpaOnnxOfflineRecognizerConfig(
modelConfig: SherpaOnnxOfflineModelConfig, modelConfig: SherpaOnnxOfflineModelConfig,
lmConfig: SherpaOnnxOfflineLMConfig = sherpaOnnxOfflineLMConfig(), lmConfig: SherpaOnnxOfflineLMConfig = sherpaOnnxOfflineLMConfig(),
decodingMethod: String = "greedy_search", decodingMethod: String = "greedy_search",
maxActivePaths: Int = 4 maxActivePaths: Int = 4,
hotwordsFile: String = "",
hotwordsScore: Float = 1.5
) -> SherpaOnnxOfflineRecognizerConfig { ) -> SherpaOnnxOfflineRecognizerConfig {
return SherpaOnnxOfflineRecognizerConfig( return SherpaOnnxOfflineRecognizerConfig(
feat_config: featConfig, feat_config: featConfig,
model_config: modelConfig, model_config: modelConfig,
lm_config: lmConfig, lm_config: lmConfig,
decoding_method: toCPointer(decodingMethod), decoding_method: toCPointer(decodingMethod),
max_active_paths: Int32(maxActivePaths) max_active_paths: Int32(maxActivePaths),
hotwords_file: toCPointer(hotwordsFile),
hotwords_score: hotwordsScore
) )
} }