Skip to content

Commit 9c46d00

Browse files
authored
Add APK for small paraformer (k2-fsa#1133)
1 parent 9ad6145 commit 9c46d00

File tree

2 files changed

+34
-0
lines changed

2 files changed

+34
-0
lines changed

scripts/apk/generate-vad-asr-apk-script.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,29 @@ def get_models():
8989
popd
9090
""",
9191
),
92+
Model(
93+
model_name="sherpa-onnx-paraformer-zh-small-2024-03-09",
94+
idx=14,
95+
lang="zh",
96+
short_name="small_paraformer",
97+
rule_fsts="itn_zh_number.fst",
98+
cmd="""
99+
if [ ! -f itn_zh_number.fst ]; then
100+
curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/itn_zh_number.fst
101+
fi
102+
pushd $model_name
103+
104+
rm -fv README.md
105+
rm -fv *.py
106+
rm -fv *.yaml
107+
rm -fv *.mvn
108+
rm -rfv test_wavs
109+
110+
ls -lh
111+
112+
popd
113+
""",
114+
),
92115
Model(
93116
model_name="icefall-asr-zipformer-wenetspeech-20230615",
94117
idx=4,

sherpa-onnx/kotlin-api/OfflineRecognizer.kt

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
310310
modelType = "transducer",
311311
)
312312
}
313+
314+
14 -> {
315+
val modelDir = "sherpa-onnx-paraformer-zh-small-2024-03-09"
316+
return OfflineModelConfig(
317+
paraformer = OfflineParaformerModelConfig(
318+
model = "$modelDir/model.int8.onnx",
319+
),
320+
tokens = "$modelDir/tokens.txt",
321+
modelType = "paraformer",
322+
)
323+
}
313324
}
314325
return null
315326
}

0 commit comments

Comments
 (0)