Spaces:
Running
Running
Commit
·
7552061
1
Parent(s):
40e123f
add 2 new models from coqui
Browse files
model.py
CHANGED
|
@@ -205,6 +205,8 @@ def _get_vits_zh_aishell3(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
| 205 |
def _get_vits_hf(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
| 206 |
if "fanchen" in repo_id or "vits-cantonese-hf-xiaomaiiwn" in repo_id:
|
| 207 |
model = repo_id.split("/")[-1]
|
|
|
|
|
|
|
| 208 |
else:
|
| 209 |
model = repo_id.split("-")[-1]
|
| 210 |
|
|
@@ -293,6 +295,9 @@ chinese_models = {
|
|
| 293 |
english_models = {
|
| 294 |
"csukuangfj/vits-vctk": _get_vits_vctk, # 109 speakers
|
| 295 |
"csukuangfj/vits-ljs": _get_vits_ljs,
|
|
|
|
|
|
|
|
|
|
| 296 |
# piper, US
|
| 297 |
"csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper,
|
| 298 |
"csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper,
|
|
|
|
| 205 |
def _get_vits_hf(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
| 206 |
if "fanchen" in repo_id or "vits-cantonese-hf-xiaomaiiwn" in repo_id:
|
| 207 |
model = repo_id.split("/")[-1]
|
| 208 |
+
elif "coqui" in repo_id:
|
| 209 |
+
model = "model"
|
| 210 |
else:
|
| 211 |
model = repo_id.split("-")[-1]
|
| 212 |
|
|
|
|
| 295 |
english_models = {
|
| 296 |
"csukuangfj/vits-vctk": _get_vits_vctk, # 109 speakers
|
| 297 |
"csukuangfj/vits-ljs": _get_vits_ljs,
|
| 298 |
+
# coqui-ai
|
| 299 |
+
"csukuangfj/vits-coqui-en-ljspeech": _get_vits_hf,
|
| 300 |
+
"csukuangfj/vits-coqui-en-ljspeech-neon": _get_vits_hf,
|
| 301 |
# piper, US
|
| 302 |
"csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper,
|
| 303 |
"csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper,
|