Fangjun Kuang
Committed by GitHub

Add APK for small paraformer (#1133)

@@ -90,6 +90,29 @@ def get_models(): @@ -90,6 +90,29 @@ def get_models():
90 """, 90 """,
91 ), 91 ),
92 Model( 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 + ),
  115 + Model(
93 model_name="icefall-asr-zipformer-wenetspeech-20230615", 116 model_name="icefall-asr-zipformer-wenetspeech-20230615",
94 idx=4, 117 idx=4,
95 lang="zh", 118 lang="zh",
@@ -310,6 +310,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? { @@ -310,6 +310,17 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
310 modelType = "transducer", 310 modelType = "transducer",
311 ) 311 )
312 } 312 }
  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 + }
313 } 324 }
314 return null 325 return null
315 } 326 }