Build Android APK for Thai (#1036)

2024-06-20 18:05:57 +08:00
parent 675fb1574f
commit 36336b31f4
3 changed files with 43 additions and 8 deletions
--- a/sherpa-onnx/kotlin-api/OfflineRecognizer.kt
+++ b/sherpa-onnx/kotlin-api/OfflineRecognizer.kt
@@ -284,6 +284,19 @@ fun getOfflineModelConfig(type: Int): OfflineModelConfig? {
                modelType = "tele_speech",
            )
        }
+
+        12 -> {
+            val modelDir = "sherpa-onnx-zipformer-thai-2024-06-20"
+            return OfflineModelConfig(
+                transducer = OfflineTransducerModelConfig(
+                    encoder = "$modelDir/encoder-epoch-12-avg-5.int8.onnx",
+                    decoder = "$modelDir/decoder-epoch-12-avg-5.onnx",
+                    joiner = "$modelDir/joiner-epoch-12-avg-5.int8.onnx",
+                ),
+                tokens = "$modelDir/tokens.txt",
+                modelType = "zipformer2",
+            )
+        }
    }
    return null
 }