Fangjun Kuang
Committed by GitHub

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
@@ -29,7 +29,7 @@ class TtsModel: @@ -29,7 +29,7 @@ class TtsModel:
29 model_dir: str 29 model_dir: str
30 model_name: str 30 model_name: str
31 lang: str # en, zh, fr, de, etc. 31 lang: str # en, zh, fr, de, etc.
32 - rule_fsts: Optional[List[str]] = (None,) 32 + rule_fsts: Optional[List[str]] = None
33 33
34 34
35 def get_all_models() -> List[TtsModel]: 35 def get_all_models() -> List[TtsModel]:
@@ -89,8 +89,13 @@ def get_all_models() -> List[TtsModel]: @@ -89,8 +89,13 @@ def get_all_models() -> List[TtsModel]:
89 lang="zh", 89 lang="zh",
90 rule_fsts="vits-zh-hf-theresa/rule.fst", 90 rule_fsts="vits-zh-hf-theresa/rule.fst",
91 ), 91 ),
92 - # English (US) 92 + # English (coqui-ai/TTS)
93 # fmt: off 93 # fmt: off
  94 + TtsModel(model_dir="vits-coqui-en-ljspeech", model_name="model.onnx", lang="en"),
  95 + TtsModel(model_dir="vits-coqui-en-ljspeech-neon", model_name="model.onnx", lang="en"),
  96 + TtsModel(model_dir="vits-coqui-en-vctk", model_name="model.onnx", lang="en"),
  97 + # TtsModel(model_dir="vits-coqui-en-jenny", model_name="model.onnx", lang="en"),
  98 + # English (US)
94 TtsModel(model_dir="vits-vctk", model_name="vits-vctk.onnx", lang="en"), 99 TtsModel(model_dir="vits-vctk", model_name="vits-vctk.onnx", lang="en"),
95 TtsModel(model_dir="vits-ljs", model_name="vits-ljs.onnx", lang="en"), 100 TtsModel(model_dir="vits-ljs", model_name="vits-ljs.onnx", lang="en"),
96 TtsModel(model_dir="vits-piper-en_US-amy-low", model_name="en_US-amy-low.onnx", lang="en",), 101 TtsModel(model_dir="vits-piper-en_US-amy-low", model_name="en_US-amy-low.onnx", lang="en",),