csukuangfj commited on
Commit
332987c
·
2 Parent(s): 9ca4b94 3ff7e82

Merge remote-tracking branch 'origin/main'

Browse files
Files changed (1) hide show
  1. model.py +13 -1
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-es-glados-medium": _get_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()),