fangjun commited on
Commit
69a4c24
·
1 Parent(s): 5d5d31a

add hindi models

Browse files
Files changed (1) hide show
  1. model.py +10 -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:
@@ -761,7 +763,8 @@ german_models = {
761
 
762
  spanish_models = {
763
  # "csukuangfj/vits-coqui-es-css10": _get_vits_piper,
764
- "csukuangfj/vits-piper-es-glados-medium": _get_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()),