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