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 sense voice (#1164)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-07-22 14:08:40 +08:00
committed by GitHub
parent ac8223bd8a
commit dd300b1de5
16 changed files with 601 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
sherpa-onnx/jni/offline-recognizer.cc
View File

@@ -171,6 +171,31 @@ static OfflineRecognizerConfig GetOfflineConfig(JNIEnv *env, jobject config) {
ans.model_config.whisper.tail_paddings =
env->GetIntField(whisper_config, fid);
// sense voice
fid = env->GetFieldID(model_config_cls, "senseVoice",
"Lcom/k2fsa/sherpa/onnx/OfflineSenseVoiceModelConfig;");
jobject sense_voice_config = env->GetObjectField(model_config, fid);
jclass sense_voice_config_cls = env->GetObjectClass(sense_voice_config);
fid = env->GetFieldID(sense_voice_config_cls, "model", "Ljava/lang/String;");
s = (jstring)env->GetObjectField(sense_voice_config, fid);
p = env->GetStringUTFChars(s, nullptr);
ans.model_config.sense_voice.model = p;
env->ReleaseStringUTFChars(s, p);
fid =
env->GetFieldID(sense_voice_config_cls, "language", "Ljava/lang/String;");
s = (jstring)env->GetObjectField(sense_voice_config, fid);
p = env->GetStringUTFChars(s, nullptr);
ans.model_config.sense_voice.language = p;
env->ReleaseStringUTFChars(s, p);
fid = env->GetFieldID(sense_voice_config_cls, "useInverseTextNormalization",
"Z");
ans.model_config.sense_voice.use_itn =
env->GetBooleanField(sense_voice_config, fid);
// nemo
fid = env->GetFieldID(
model_config_cls, "nemo",
"Lcom/k2fsa/sherpa/onnx/OfflineNemoEncDecCtcModelConfig;");