csukuangfj commited on
Commit
913005c
·
1 Parent(s): 4ca6202

add two new models

Browse files
Files changed (1) hide show
  1. model.py +8 -0
model.py CHANGED
@@ -634,6 +634,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
634
  return hungarian_models[repo_id](repo_id, speed)
635
  elif repo_id in icelandic_models:
636
  return icelandic_models[repo_id](repo_id, speed)
 
 
637
  elif repo_id in italian_models:
638
  return italian_models[repo_id](repo_id, speed)
639
  elif repo_id in georgian_models:
@@ -895,6 +897,11 @@ finnish_models = {
895
  hindi_models = {
896
  "csukuangfj/vits-piper-hi_IN-pratham-medium": _get_vits_piper,
897
  "csukuangfj/vits-piper-hi_IN-priyamvada-medium": _get_vits_piper,
 
 
 
 
 
898
  }
899
 
900
  hungarian_models = {
@@ -1117,6 +1124,7 @@ language_to_models = {
1117
  "Hindi": list(hindi_models.keys()),
1118
  "Hungarian": list(hungarian_models.keys()),
1119
  "Icelandic": list(icelandic_models.keys()),
 
1120
  "Irish": list(irish_models.keys()),
1121
  "Italian": list(italian_models.keys()),
1122
  "Kazakh": list(kazakh_models.keys()),
 
634
  return hungarian_models[repo_id](repo_id, speed)
635
  elif repo_id in icelandic_models:
636
  return icelandic_models[repo_id](repo_id, speed)
637
+ elif repo_id in indonesian_models:
638
+ return indonesian_models[repo_id](repo_id, speed)
639
  elif repo_id in italian_models:
640
  return italian_models[repo_id](repo_id, speed)
641
  elif repo_id in georgian_models:
 
897
  hindi_models = {
898
  "csukuangfj/vits-piper-hi_IN-pratham-medium": _get_vits_piper,
899
  "csukuangfj/vits-piper-hi_IN-priyamvada-medium": _get_vits_piper,
900
+ "csukuangfj/vits-piper-hi_IN-rohan-medium": _get_vits_piper,
901
+ }
902
+
903
+ indonesian_models = {
904
+ "csukuangfj/vits-piper-id_ID-news_tts-medium": _get_vits_piper,
905
  }
906
 
907
  hungarian_models = {
 
1124
  "Hindi": list(hindi_models.keys()),
1125
  "Hungarian": list(hungarian_models.keys()),
1126
  "Icelandic": list(icelandic_models.keys()),
1127
+ "Indonesian": list(indonesian_models.keys()),
1128
  "Irish": list(irish_models.keys()),
1129
  "Italian": list(italian_models.keys()),
1130
  "Kazakh": list(kazakh_models.keys()),