From 3c1ea990b1e0d198367213350f914d5f4d40f572 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 11 Nov 2023 13:27:15 +0800 Subject: [PATCH] Build Android APKs for VITS models from Coqui-ai/TTS (#419) Support building Android APKs for VITS models from https://github.com/coqui-ai/TTS Please see https://k2-fsa.github.io/sherpa/onnx/tts/apk.html --- scripts/apk/generate-tts-apk-script.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/scripts/apk/generate-tts-apk-script.py b/scripts/apk/generate-tts-apk-script.py index 43df6535a..a9f94ae61 100755 --- a/scripts/apk/generate-tts-apk-script.py +++ b/scripts/apk/generate-tts-apk-script.py @@ -29,7 +29,7 @@ class TtsModel: model_dir: str model_name: str lang: str # en, zh, fr, de, etc. - rule_fsts: Optional[List[str]] = (None,) + rule_fsts: Optional[List[str]] = None def get_all_models() -> List[TtsModel]: @@ -89,8 +89,13 @@ def get_all_models() -> List[TtsModel]: lang="zh", rule_fsts="vits-zh-hf-theresa/rule.fst", ), - # English (US) + # English (coqui-ai/TTS) # fmt: off + TtsModel(model_dir="vits-coqui-en-ljspeech", model_name="model.onnx", lang="en"), + TtsModel(model_dir="vits-coqui-en-ljspeech-neon", model_name="model.onnx", lang="en"), + TtsModel(model_dir="vits-coqui-en-vctk", model_name="model.onnx", lang="en"), + # TtsModel(model_dir="vits-coqui-en-jenny", model_name="model.onnx", lang="en"), + # English (US) TtsModel(model_dir="vits-vctk", model_name="vits-vctk.onnx", lang="en"), TtsModel(model_dir="vits-ljs", model_name="vits-ljs.onnx", lang="en"), TtsModel(model_dir="vits-piper-en_US-amy-low", model_name="en_US-amy-low.onnx", lang="en",),