Spaces:
Runtime error
Runtime error
csukuangfj
commited on
Commit
•
af3a548
1
Parent(s):
c573ca8
add more models
Browse files
model.py
CHANGED
@@ -357,6 +357,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
357 |
return korean_models[repo_id](repo_id, speed)
|
358 |
elif repo_id in afrikaans_models:
|
359 |
return afrikaans_models[repo_id](repo_id, speed)
|
|
|
|
|
360 |
else:
|
361 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
362 |
|
@@ -445,6 +447,7 @@ spanish_models = {
|
|
445 |
"csukuangfj/vits-piper-es_ES-mls_9972-low": _get_vits_piper,
|
446 |
"csukuangfj/vits-piper-es_ES-sharvard-medium": _get_vits_piper, # 2 speakers
|
447 |
"csukuangfj/vits-piper-es_MX-ald-medium": _get_vits_piper,
|
|
|
448 |
}
|
449 |
|
450 |
french_models = {
|
@@ -503,6 +506,7 @@ finnish_models = {
|
|
503 |
"csukuangfj/vits-coqui-fi-css10": _get_vits_piper,
|
504 |
"csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
|
505 |
"csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
|
|
|
506 |
}
|
507 |
|
508 |
hungarian_models = {
|
@@ -638,6 +642,7 @@ slovenian_models = {
|
|
638 |
"csukuangfj/vits-coqui-sl-cv": _get_vits_piper,
|
639 |
}
|
640 |
|
|
|
641 |
bengali_models = {
|
642 |
"csukuangfj/vits-coqui-bn-custom_female": _get_vits_piper,
|
643 |
"csukuangfj/vits-mimic3-bn-multi_low": _get_vits_piper,
|
@@ -664,6 +669,10 @@ afrikaans_models = {
|
|
664 |
"csukuangfj/vits-mimic3-af_ZA-google-nwu_low": _get_vits_piper,
|
665 |
}
|
666 |
|
|
|
|
|
|
|
|
|
667 |
|
668 |
language_to_models = {
|
669 |
"English": list(english_models.keys()),
|
@@ -685,6 +694,7 @@ language_to_models = {
|
|
685 |
"Georgian": list(georgian_models.keys()),
|
686 |
"German": list(german_models.keys()),
|
687 |
"Greek": list(greek_models.keys()),
|
|
|
688 |
"Hungarian": list(hungarian_models.keys()),
|
689 |
"Icelandic": list(icelandic_models.keys()),
|
690 |
"Irish": list(irish_models.keys()),
|
|
|
357 |
return korean_models[repo_id](repo_id, speed)
|
358 |
elif repo_id in afrikaans_models:
|
359 |
return afrikaans_models[repo_id](repo_id, speed)
|
360 |
+
elif repo_id in gujarati_models:
|
361 |
+
return gujarati_models[repo_id](repo_id, speed)
|
362 |
else:
|
363 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
364 |
|
|
|
447 |
"csukuangfj/vits-piper-es_ES-mls_9972-low": _get_vits_piper,
|
448 |
"csukuangfj/vits-piper-es_ES-sharvard-medium": _get_vits_piper, # 2 speakers
|
449 |
"csukuangfj/vits-piper-es_MX-ald-medium": _get_vits_piper,
|
450 |
+
"csukuangfj/vits-mimic3-es_ES-m-ailabs_low": _get_vits_piper,
|
451 |
}
|
452 |
|
453 |
french_models = {
|
|
|
506 |
"csukuangfj/vits-coqui-fi-css10": _get_vits_piper,
|
507 |
"csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
|
508 |
"csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
|
509 |
+
"csukuangfj/vits-mimic3-fi_FI-harri-tapani-ylilammi_low": _get_vits_piper,
|
510 |
}
|
511 |
|
512 |
hungarian_models = {
|
|
|
642 |
"csukuangfj/vits-coqui-sl-cv": _get_vits_piper,
|
643 |
}
|
644 |
|
645 |
+
# Bangla
|
646 |
bengali_models = {
|
647 |
"csukuangfj/vits-coqui-bn-custom_female": _get_vits_piper,
|
648 |
"csukuangfj/vits-mimic3-bn-multi_low": _get_vits_piper,
|
|
|
669 |
"csukuangfj/vits-mimic3-af_ZA-google-nwu_low": _get_vits_piper,
|
670 |
}
|
671 |
|
672 |
+
gujarati_models = {
|
673 |
+
"csukuangfj/vits-mimic3-gu_IN-cmu-indic_low": _get_vits_piper,
|
674 |
+
}
|
675 |
+
|
676 |
|
677 |
language_to_models = {
|
678 |
"English": list(english_models.keys()),
|
|
|
694 |
"Georgian": list(georgian_models.keys()),
|
695 |
"German": list(german_models.keys()),
|
696 |
"Greek": list(greek_models.keys()),
|
697 |
+
"Gujarati": list(gujarati_models.keys()),
|
698 |
"Hungarian": list(hungarian_models.keys()),
|
699 |
"Icelandic": list(icelandic_models.keys()),
|
700 |
"Irish": list(irish_models.keys()),
|