Spaces:
Running
Running
csukuangfj
commited on
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,
|