Add non-streaming zipformer Android APK (#1052)

2024-06-24 16:22:19 +08:00
parent e7a45108ac
commit 1f95bff719
2 changed files with 34 additions and 0 deletions
--- a/sherpa-onnx/kotlin-api/OfflineRecognizer.kt
+++ b/sherpa-onnx/kotlin-api/OfflineRecognizer.kt
@@ -297,6 +297,19 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
                modelType = "zipformer2",
            )
        }
+
+        13 -> {
+            val modelDir = "sherpa-onnx-zipformer-korean-2024-06-24"
+            return OfflineModelConfig(
+                transducer = OfflineTransducerModelConfig(
+                    encoder = "$modelDir/encoder-epoch-99-avg-1.int8.onnx",
+                    decoder = "$modelDir/decoder-epoch-99-avg-1.onnx",
+                    joiner = "$modelDir/joiner-epoch-99-avg-1.int8.onnx",
+                ),
+                tokens = "$modelDir/tokens.txt",
+                modelType = "zipformer2",
+            )
+        }
    }
    return null
 }