csukuangfj commited on
Commit
af3a548
1 Parent(s): c573ca8

add more models

Browse files
Files changed (1) hide show
  1. model.py +10 -0
model.py CHANGED
@@ -357,6 +357,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
357
  return korean_models[repo_id](repo_id, speed)
358
  elif repo_id in afrikaans_models:
359
  return afrikaans_models[repo_id](repo_id, speed)
 
 
360
  else:
361
  raise ValueError(f"Unsupported repo_id: {repo_id}")
362
 
@@ -445,6 +447,7 @@ spanish_models = {
445
  "csukuangfj/vits-piper-es_ES-mls_9972-low": _get_vits_piper,
446
  "csukuangfj/vits-piper-es_ES-sharvard-medium": _get_vits_piper, # 2 speakers
447
  "csukuangfj/vits-piper-es_MX-ald-medium": _get_vits_piper,
 
448
  }
449
 
450
  french_models = {
@@ -503,6 +506,7 @@ finnish_models = {
503
  "csukuangfj/vits-coqui-fi-css10": _get_vits_piper,
504
  "csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
505
  "csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
 
506
  }
507
 
508
  hungarian_models = {
@@ -638,6 +642,7 @@ slovenian_models = {
638
  "csukuangfj/vits-coqui-sl-cv": _get_vits_piper,
639
  }
640
 
 
641
  bengali_models = {
642
  "csukuangfj/vits-coqui-bn-custom_female": _get_vits_piper,
643
  "csukuangfj/vits-mimic3-bn-multi_low": _get_vits_piper,
@@ -664,6 +669,10 @@ afrikaans_models = {
664
  "csukuangfj/vits-mimic3-af_ZA-google-nwu_low": _get_vits_piper,
665
  }
666
 
 
 
 
 
667
 
668
  language_to_models = {
669
  "English": list(english_models.keys()),
@@ -685,6 +694,7 @@ language_to_models = {
685
  "Georgian": list(georgian_models.keys()),
686
  "German": list(german_models.keys()),
687
  "Greek": list(greek_models.keys()),
 
688
  "Hungarian": list(hungarian_models.keys()),
689
  "Icelandic": list(icelandic_models.keys()),
690
  "Irish": list(irish_models.keys()),
 
357
  return korean_models[repo_id](repo_id, speed)
358
  elif repo_id in afrikaans_models:
359
  return afrikaans_models[repo_id](repo_id, speed)
360
+ elif repo_id in gujarati_models:
361
+ return gujarati_models[repo_id](repo_id, speed)
362
  else:
363
  raise ValueError(f"Unsupported repo_id: {repo_id}")
364
 
 
447
  "csukuangfj/vits-piper-es_ES-mls_9972-low": _get_vits_piper,
448
  "csukuangfj/vits-piper-es_ES-sharvard-medium": _get_vits_piper, # 2 speakers
449
  "csukuangfj/vits-piper-es_MX-ald-medium": _get_vits_piper,
450
+ "csukuangfj/vits-mimic3-es_ES-m-ailabs_low": _get_vits_piper,
451
  }
452
 
453
  french_models = {
 
506
  "csukuangfj/vits-coqui-fi-css10": _get_vits_piper,
507
  "csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
508
  "csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
509
+ "csukuangfj/vits-mimic3-fi_FI-harri-tapani-ylilammi_low": _get_vits_piper,
510
  }
511
 
512
  hungarian_models = {
 
642
  "csukuangfj/vits-coqui-sl-cv": _get_vits_piper,
643
  }
644
 
645
+ # Bangla
646
  bengali_models = {
647
  "csukuangfj/vits-coqui-bn-custom_female": _get_vits_piper,
648
  "csukuangfj/vits-mimic3-bn-multi_low": _get_vits_piper,
 
669
  "csukuangfj/vits-mimic3-af_ZA-google-nwu_low": _get_vits_piper,
670
  }
671
 
672
+ gujarati_models = {
673
+ "csukuangfj/vits-mimic3-gu_IN-cmu-indic_low": _get_vits_piper,
674
+ }
675
+
676
 
677
  language_to_models = {
678
  "English": list(english_models.keys()),
 
694
  "Georgian": list(georgian_models.keys()),
695
  "German": list(german_models.keys()),
696
  "Greek": list(greek_models.keys()),
697
+ "Gujarati": list(gujarati_models.keys()),
698
  "Hungarian": list(hungarian_models.keys()),
699
  "Icelandic": list(icelandic_models.keys()),
700
  "Irish": list(irish_models.keys()),