Spaces:
Runtime error
Runtime error
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()),
|