Spaces:
Running
Running
fangjun
commited on
Commit
·
69a4c24
1
Parent(s):
5d5d31a
add hindi models
Browse files
model.py
CHANGED
|
@@ -580,6 +580,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
| 580 |
return greek_models[repo_id](repo_id, speed)
|
| 581 |
elif repo_id in finnish_models:
|
| 582 |
return finnish_models[repo_id](repo_id, speed)
|
|
|
|
|
|
|
| 583 |
elif repo_id in hungarian_models:
|
| 584 |
return hungarian_models[repo_id](repo_id, speed)
|
| 585 |
elif repo_id in icelandic_models:
|
|
@@ -761,7 +763,8 @@ german_models = {
|
|
| 761 |
|
| 762 |
spanish_models = {
|
| 763 |
# "csukuangfj/vits-coqui-es-css10": _get_vits_piper,
|
| 764 |
-
"csukuangfj/vits-piper-
|
|
|
|
| 765 |
"csukuangfj/vits-piper-es_ES-carlfm-x_low": _get_vits_piper,
|
| 766 |
"csukuangfj/vits-piper-es_ES-davefx-medium": _get_vits_piper,
|
| 767 |
# "csukuangfj/vits-piper-es_ES-mls_10246-low": _get_vits_piper,
|
|
@@ -833,6 +836,11 @@ finnish_models = {
|
|
| 833 |
"csukuangfj/vits-mimic3-fi_FI-harri-tapani-ylilammi_low": _get_vits_piper,
|
| 834 |
}
|
| 835 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 836 |
hungarian_models = {
|
| 837 |
# "csukuangfj/vits-coqui-hu-css10": _get_vits_piper,
|
| 838 |
"csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
|
|
@@ -1039,6 +1047,7 @@ language_to_models = {
|
|
| 1039 |
"German": list(german_models.keys()),
|
| 1040 |
"Greek": list(greek_models.keys()),
|
| 1041 |
"Gujarati": list(gujarati_models.keys()),
|
|
|
|
| 1042 |
"Hungarian": list(hungarian_models.keys()),
|
| 1043 |
"Icelandic": list(icelandic_models.keys()),
|
| 1044 |
"Irish": list(irish_models.keys()),
|
|
|
|
| 580 |
return greek_models[repo_id](repo_id, speed)
|
| 581 |
elif repo_id in finnish_models:
|
| 582 |
return finnish_models[repo_id](repo_id, speed)
|
| 583 |
+
elif repo_id in hindi_models:
|
| 584 |
+
return hindi_models[repo_id](repo_id, speed)
|
| 585 |
elif repo_id in hungarian_models:
|
| 586 |
return hungarian_models[repo_id](repo_id, speed)
|
| 587 |
elif repo_id in icelandic_models:
|
|
|
|
| 763 |
|
| 764 |
spanish_models = {
|
| 765 |
# "csukuangfj/vits-coqui-es-css10": _get_vits_piper,
|
| 766 |
+
"csukuangfj/vits-piper-es_ES-glados-medium": _get_vits_piper,
|
| 767 |
+
"csukuangfj/vits-piper-es_AR-daniela-high": _get_vits_piper,
|
| 768 |
"csukuangfj/vits-piper-es_ES-carlfm-x_low": _get_vits_piper,
|
| 769 |
"csukuangfj/vits-piper-es_ES-davefx-medium": _get_vits_piper,
|
| 770 |
# "csukuangfj/vits-piper-es_ES-mls_10246-low": _get_vits_piper,
|
|
|
|
| 836 |
"csukuangfj/vits-mimic3-fi_FI-harri-tapani-ylilammi_low": _get_vits_piper,
|
| 837 |
}
|
| 838 |
|
| 839 |
+
hindi_models = {
|
| 840 |
+
"csukuangfj/vits-piper-hi_IN-pratham-medium": _get_vits_piper,
|
| 841 |
+
"csukuangfj/vits-piper-hi_IN-priyamvada-medium": _get_vits_piper,
|
| 842 |
+
}
|
| 843 |
+
|
| 844 |
hungarian_models = {
|
| 845 |
# "csukuangfj/vits-coqui-hu-css10": _get_vits_piper,
|
| 846 |
"csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
|
|
|
|
| 1047 |
"German": list(german_models.keys()),
|
| 1048 |
"Greek": list(greek_models.keys()),
|
| 1049 |
"Gujarati": list(gujarati_models.keys()),
|
| 1050 |
+
"Hindi": list(hindi_models.keys()),
|
| 1051 |
"Hungarian": list(hungarian_models.keys()),
|
| 1052 |
"Icelandic": list(icelandic_models.keys()),
|
| 1053 |
"Irish": list(irish_models.keys()),
|