Spaces:
Running
Running
Commit
·
71c0d81
1
Parent(s):
3d1c4ab
add more models
Browse files
model.py
CHANGED
|
@@ -414,6 +414,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
| 414 |
return gujarati_models[repo_id](repo_id, speed)
|
| 415 |
elif repo_id in tswana_models:
|
| 416 |
return tswana_models[repo_id](repo_id, speed)
|
|
|
|
|
|
|
| 417 |
else:
|
| 418 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
| 419 |
|
|
@@ -446,6 +448,9 @@ english_models = {
|
|
| 446 |
"csukuangfj/vits-piper-en_US-glados|1 speaker": _get_vits_piper,
|
| 447 |
"csukuangfj/vits-piper-en_GB-southern_english_male-medium|8 speakers": _get_vits_piper,
|
| 448 |
"csukuangfj/vits-piper-en_GB-southern_english_female-medium|6 speakers": _get_vits_piper,
|
|
|
|
|
|
|
|
|
|
| 449 |
# coqui-ai
|
| 450 |
"csukuangfj/vits-coqui-en-ljspeech|1 speaker": _get_vits_piper,
|
| 451 |
"csukuangfj/vits-coqui-en-ljspeech-neon|1 speaker": _get_vits_piper,
|
|
@@ -457,6 +462,7 @@ english_models = {
|
|
| 457 |
"csukuangfj/vits-piper-en_US-arctic-medium|18 speakers": _get_vits_piper, # 18 speakers
|
| 458 |
"csukuangfj/vits-piper-en_US-danny-low|1 speaker": _get_vits_piper,
|
| 459 |
"csukuangfj/vits-piper-en_US-hfc_male-medium|1 speaker": _get_vits_piper,
|
|
|
|
| 460 |
"csukuangfj/vits-piper-en_US-joe-medium|1 speaker": _get_vits_piper,
|
| 461 |
"csukuangfj/vits-piper-en_US-kathleen-low|1 speaker": _get_vits_piper,
|
| 462 |
"csukuangfj/vits-piper-en_US-kusal-medium|1 speaker": _get_vits_piper,
|
|
@@ -518,8 +524,9 @@ french_models = {
|
|
| 518 |
"csukuangfj/vits-coqui-fr-css10": _get_vits_piper,
|
| 519 |
# "csukuangfj/vits-piper-fr_FR-gilles-low": _get_vits_piper,
|
| 520 |
# "csukuangfj/vits-piper-fr_FR-mls_1840-low": _get_vits_piper,
|
| 521 |
-
"csukuangfj/vits-piper-fr_FR-mls-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
|
| 522 |
"csukuangfj/vits-piper-fr_FR-upmc-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
|
|
|
|
| 523 |
"csukuangfj/vits-piper-fr_FR-siwis-low": _get_vits_piper, # female
|
| 524 |
"csukuangfj/vits-piper-fr_FR-siwis-medium": _get_vits_piper,
|
| 525 |
"csukuangfj/vits-piper-fr_FR-tjiho-model1": _get_vits_piper,
|
|
@@ -591,6 +598,7 @@ icelandic_models = {
|
|
| 591 |
|
| 592 |
italian_models = {
|
| 593 |
"csukuangfj/vits-piper-it_IT-riccardo-x_low": _get_vits_piper,
|
|
|
|
| 594 |
}
|
| 595 |
|
| 596 |
georgian_models = {
|
|
@@ -671,6 +679,7 @@ swahili_models = {
|
|
| 671 |
turkish_models = {
|
| 672 |
"csukuangfj/vits-piper-tr_TR-dfki-medium": _get_vits_piper,
|
| 673 |
"csukuangfj/vits-piper-tr_TR-fahrettin-medium": _get_vits_piper,
|
|
|
|
| 674 |
}
|
| 675 |
|
| 676 |
vietnamese_models = {
|
|
@@ -750,6 +759,9 @@ tswana_models = {
|
|
| 750 |
"csukuangfj/vits-mimic3-tn_ZA-google-nwu_low": _get_vits_piper,
|
| 751 |
}
|
| 752 |
|
|
|
|
|
|
|
|
|
|
| 753 |
|
| 754 |
language_to_models = {
|
| 755 |
"English": list(english_models.keys()),
|
|
@@ -800,4 +812,5 @@ language_to_models = {
|
|
| 800 |
"Turkish": list(turkish_models.keys()),
|
| 801 |
"Ukrainian": list(ukrainian_models.keys()),
|
| 802 |
"Vietnamese": list(vietnamese_models.keys()),
|
|
|
|
| 803 |
}
|
|
|
|
| 414 |
return gujarati_models[repo_id](repo_id, speed)
|
| 415 |
elif repo_id in tswana_models:
|
| 416 |
return tswana_models[repo_id](repo_id, speed)
|
| 417 |
+
elif repo_id in welsh_models:
|
| 418 |
+
return welsh_models[repo_id](repo_id, speed)
|
| 419 |
else:
|
| 420 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
| 421 |
|
|
|
|
| 448 |
"csukuangfj/vits-piper-en_US-glados|1 speaker": _get_vits_piper,
|
| 449 |
"csukuangfj/vits-piper-en_GB-southern_english_male-medium|8 speakers": _get_vits_piper,
|
| 450 |
"csukuangfj/vits-piper-en_GB-southern_english_female-medium|6 speakers": _get_vits_piper,
|
| 451 |
+
"csukuangfj/vits-piper-en_US-bryce-medium|1 speaker": _get_vits_piper,
|
| 452 |
+
"csukuangfj/vits-piper-en_US-john-medium|1 speaker": _get_vits_piper,
|
| 453 |
+
"csukuangfj/vits-piper-en_US-norman-medium|1 speaker": _get_vits_piper,
|
| 454 |
# coqui-ai
|
| 455 |
"csukuangfj/vits-coqui-en-ljspeech|1 speaker": _get_vits_piper,
|
| 456 |
"csukuangfj/vits-coqui-en-ljspeech-neon|1 speaker": _get_vits_piper,
|
|
|
|
| 462 |
"csukuangfj/vits-piper-en_US-arctic-medium|18 speakers": _get_vits_piper, # 18 speakers
|
| 463 |
"csukuangfj/vits-piper-en_US-danny-low|1 speaker": _get_vits_piper,
|
| 464 |
"csukuangfj/vits-piper-en_US-hfc_male-medium|1 speaker": _get_vits_piper,
|
| 465 |
+
"csukuangfj/vits-piper-en_US-hfc_female-medium|1 speaker": _get_vits_piper,
|
| 466 |
"csukuangfj/vits-piper-en_US-joe-medium|1 speaker": _get_vits_piper,
|
| 467 |
"csukuangfj/vits-piper-en_US-kathleen-low|1 speaker": _get_vits_piper,
|
| 468 |
"csukuangfj/vits-piper-en_US-kusal-medium|1 speaker": _get_vits_piper,
|
|
|
|
| 524 |
"csukuangfj/vits-coqui-fr-css10": _get_vits_piper,
|
| 525 |
# "csukuangfj/vits-piper-fr_FR-gilles-low": _get_vits_piper,
|
| 526 |
# "csukuangfj/vits-piper-fr_FR-mls_1840-low": _get_vits_piper,
|
| 527 |
+
# "csukuangfj/vits-piper-fr_FR-mls-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
|
| 528 |
"csukuangfj/vits-piper-fr_FR-upmc-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
|
| 529 |
+
"csukuangfj/vits-piper-fr_FR-tom-medium|1 speaker": _get_vits_piper, # 2 speakers, 0-femal, 1-male
|
| 530 |
"csukuangfj/vits-piper-fr_FR-siwis-low": _get_vits_piper, # female
|
| 531 |
"csukuangfj/vits-piper-fr_FR-siwis-medium": _get_vits_piper,
|
| 532 |
"csukuangfj/vits-piper-fr_FR-tjiho-model1": _get_vits_piper,
|
|
|
|
| 598 |
|
| 599 |
italian_models = {
|
| 600 |
"csukuangfj/vits-piper-it_IT-riccardo-x_low": _get_vits_piper,
|
| 601 |
+
"csukuangfj/vits-piper-it_IT-paola-medium": _get_vits_piper,
|
| 602 |
}
|
| 603 |
|
| 604 |
georgian_models = {
|
|
|
|
| 679 |
turkish_models = {
|
| 680 |
"csukuangfj/vits-piper-tr_TR-dfki-medium": _get_vits_piper,
|
| 681 |
"csukuangfj/vits-piper-tr_TR-fahrettin-medium": _get_vits_piper,
|
| 682 |
+
"csukuangfj/vits-piper-tr_TR-fettah-medium|1 speaker": _get_vits_piper,
|
| 683 |
}
|
| 684 |
|
| 685 |
vietnamese_models = {
|
|
|
|
| 759 |
"csukuangfj/vits-mimic3-tn_ZA-google-nwu_low": _get_vits_piper,
|
| 760 |
}
|
| 761 |
|
| 762 |
+
welsh_models = {
|
| 763 |
+
"csukuangfj/vits-piper-cy_GB-gwryw_gogleddol-medium|1 speaker": _get_vits_piper,
|
| 764 |
+
}
|
| 765 |
|
| 766 |
language_to_models = {
|
| 767 |
"English": list(english_models.keys()),
|
|
|
|
| 812 |
"Turkish": list(turkish_models.keys()),
|
| 813 |
"Ukrainian": list(ukrainian_models.keys()),
|
| 814 |
"Vietnamese": list(vietnamese_models.keys()),
|
| 815 |
+
"Welsh": list(welsh_models.keys()),
|
| 816 |
}
|