Spaces:
Running
Running
Merge remote-tracking branch 'origin/main'
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:
|
|
@@ -762,7 +764,8 @@ german_models = {
|
|
| 762 |
|
| 763 |
spanish_models = {
|
| 764 |
# "csukuangfj/vits-coqui-es-css10": _get_vits_piper,
|
| 765 |
-
"csukuangfj/vits-piper-
|
|
|
|
| 766 |
"csukuangfj/vits-piper-es_ES-carlfm-x_low": _get_vits_piper,
|
| 767 |
"csukuangfj/vits-piper-es_ES-davefx-medium": _get_vits_piper,
|
| 768 |
# "csukuangfj/vits-piper-es_ES-mls_10246-low": _get_vits_piper,
|
|
@@ -834,6 +837,11 @@ finnish_models = {
|
|
| 834 |
"csukuangfj/vits-mimic3-fi_FI-harri-tapani-ylilammi_low": _get_vits_piper,
|
| 835 |
}
|
| 836 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 837 |
hungarian_models = {
|
| 838 |
# "csukuangfj/vits-coqui-hu-css10": _get_vits_piper,
|
| 839 |
"csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
|
|
@@ -992,7 +1000,10 @@ thai_models = {
|
|
| 992 |
|
| 993 |
persian_models = {
|
| 994 |
"csukuangfj/vits-piper-fa_IR-amir-medium": _get_vits_piper,
|
|
|
|
|
|
|
| 995 |
"csukuangfj/vits-piper-fa_IR-gyro-medium": _get_vits_piper,
|
|
|
|
| 996 |
"csukuangfj/vits-mimic3-fa-haaniye_low": _get_vits_piper,
|
| 997 |
}
|
| 998 |
|
|
@@ -1040,6 +1051,7 @@ language_to_models = {
|
|
| 1040 |
"German": list(german_models.keys()),
|
| 1041 |
"Greek": list(greek_models.keys()),
|
| 1042 |
"Gujarati": list(gujarati_models.keys()),
|
|
|
|
| 1043 |
"Hungarian": list(hungarian_models.keys()),
|
| 1044 |
"Icelandic": list(icelandic_models.keys()),
|
| 1045 |
"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:
|
|
|
|
| 764 |
|
| 765 |
spanish_models = {
|
| 766 |
# "csukuangfj/vits-coqui-es-css10": _get_vits_piper,
|
| 767 |
+
"csukuangfj/vits-piper-es_ES-glados-medium": _get_vits_piper,
|
| 768 |
+
"csukuangfj/vits-piper-es_AR-daniela-high": _get_vits_piper,
|
| 769 |
"csukuangfj/vits-piper-es_ES-carlfm-x_low": _get_vits_piper,
|
| 770 |
"csukuangfj/vits-piper-es_ES-davefx-medium": _get_vits_piper,
|
| 771 |
# "csukuangfj/vits-piper-es_ES-mls_10246-low": _get_vits_piper,
|
|
|
|
| 837 |
"csukuangfj/vits-mimic3-fi_FI-harri-tapani-ylilammi_low": _get_vits_piper,
|
| 838 |
}
|
| 839 |
|
| 840 |
+
hindi_models = {
|
| 841 |
+
"csukuangfj/vits-piper-hi_IN-pratham-medium": _get_vits_piper,
|
| 842 |
+
"csukuangfj/vits-piper-hi_IN-priyamvada-medium": _get_vits_piper,
|
| 843 |
+
}
|
| 844 |
+
|
| 845 |
hungarian_models = {
|
| 846 |
# "csukuangfj/vits-coqui-hu-css10": _get_vits_piper,
|
| 847 |
"csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
|
|
|
|
| 1000 |
|
| 1001 |
persian_models = {
|
| 1002 |
"csukuangfj/vits-piper-fa_IR-amir-medium": _get_vits_piper,
|
| 1003 |
+
"csukuangfj/vits-piper-fa_IR-ganji-medium": _get_vits_piper,
|
| 1004 |
+
"csukuangfj/vits-piper-fa_IR-ganji_adabi-medium": _get_vits_piper,
|
| 1005 |
"csukuangfj/vits-piper-fa_IR-gyro-medium": _get_vits_piper,
|
| 1006 |
+
"csukuangfj/vits-piper-fa_IR-reza_ibrahim-medium": _get_vits_piper,
|
| 1007 |
"csukuangfj/vits-mimic3-fa-haaniye_low": _get_vits_piper,
|
| 1008 |
}
|
| 1009 |
|
|
|
|
| 1051 |
"German": list(german_models.keys()),
|
| 1052 |
"Greek": list(greek_models.keys()),
|
| 1053 |
"Gujarati": list(gujarati_models.keys()),
|
| 1054 |
+
"Hindi": list(hindi_models.keys()),
|
| 1055 |
"Hungarian": list(hungarian_models.keys()),
|
| 1056 |
"Icelandic": list(icelandic_models.keys()),
|
| 1057 |
"Irish": list(irish_models.keys()),
|