csukuangfj commited on
Commit
71c0d81
1 Parent(s): 3d1c4ab

add more models

Browse files
Files changed (1) hide show
  1. model.py +14 -1
model.py CHANGED
@@ -414,6 +414,8 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
414
  return gujarati_models[repo_id](repo_id, speed)
415
  elif repo_id in tswana_models:
416
  return tswana_models[repo_id](repo_id, speed)
 
 
417
  else:
418
  raise ValueError(f"Unsupported repo_id: {repo_id}")
419
 
@@ -446,6 +448,9 @@ english_models = {
446
  "csukuangfj/vits-piper-en_US-glados|1 speaker": _get_vits_piper,
447
  "csukuangfj/vits-piper-en_GB-southern_english_male-medium|8 speakers": _get_vits_piper,
448
  "csukuangfj/vits-piper-en_GB-southern_english_female-medium|6 speakers": _get_vits_piper,
 
 
 
449
  # coqui-ai
450
  "csukuangfj/vits-coqui-en-ljspeech|1 speaker": _get_vits_piper,
451
  "csukuangfj/vits-coqui-en-ljspeech-neon|1 speaker": _get_vits_piper,
@@ -457,6 +462,7 @@ english_models = {
457
  "csukuangfj/vits-piper-en_US-arctic-medium|18 speakers": _get_vits_piper, # 18 speakers
458
  "csukuangfj/vits-piper-en_US-danny-low|1 speaker": _get_vits_piper,
459
  "csukuangfj/vits-piper-en_US-hfc_male-medium|1 speaker": _get_vits_piper,
 
460
  "csukuangfj/vits-piper-en_US-joe-medium|1 speaker": _get_vits_piper,
461
  "csukuangfj/vits-piper-en_US-kathleen-low|1 speaker": _get_vits_piper,
462
  "csukuangfj/vits-piper-en_US-kusal-medium|1 speaker": _get_vits_piper,
@@ -518,8 +524,9 @@ french_models = {
518
  "csukuangfj/vits-coqui-fr-css10": _get_vits_piper,
519
  # "csukuangfj/vits-piper-fr_FR-gilles-low": _get_vits_piper,
520
  # "csukuangfj/vits-piper-fr_FR-mls_1840-low": _get_vits_piper,
521
- "csukuangfj/vits-piper-fr_FR-mls-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
522
  "csukuangfj/vits-piper-fr_FR-upmc-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
 
523
  "csukuangfj/vits-piper-fr_FR-siwis-low": _get_vits_piper, # female
524
  "csukuangfj/vits-piper-fr_FR-siwis-medium": _get_vits_piper,
525
  "csukuangfj/vits-piper-fr_FR-tjiho-model1": _get_vits_piper,
@@ -591,6 +598,7 @@ icelandic_models = {
591
 
592
  italian_models = {
593
  "csukuangfj/vits-piper-it_IT-riccardo-x_low": _get_vits_piper,
 
594
  }
595
 
596
  georgian_models = {
@@ -671,6 +679,7 @@ swahili_models = {
671
  turkish_models = {
672
  "csukuangfj/vits-piper-tr_TR-dfki-medium": _get_vits_piper,
673
  "csukuangfj/vits-piper-tr_TR-fahrettin-medium": _get_vits_piper,
 
674
  }
675
 
676
  vietnamese_models = {
@@ -750,6 +759,9 @@ tswana_models = {
750
  "csukuangfj/vits-mimic3-tn_ZA-google-nwu_low": _get_vits_piper,
751
  }
752
 
 
 
 
753
 
754
  language_to_models = {
755
  "English": list(english_models.keys()),
@@ -800,4 +812,5 @@ language_to_models = {
800
  "Turkish": list(turkish_models.keys()),
801
  "Ukrainian": list(ukrainian_models.keys()),
802
  "Vietnamese": list(vietnamese_models.keys()),
 
803
  }
 
414
  return gujarati_models[repo_id](repo_id, speed)
415
  elif repo_id in tswana_models:
416
  return tswana_models[repo_id](repo_id, speed)
417
+ elif repo_id in welsh_models:
418
+ return welsh_models[repo_id](repo_id, speed)
419
  else:
420
  raise ValueError(f"Unsupported repo_id: {repo_id}")
421
 
 
448
  "csukuangfj/vits-piper-en_US-glados|1 speaker": _get_vits_piper,
449
  "csukuangfj/vits-piper-en_GB-southern_english_male-medium|8 speakers": _get_vits_piper,
450
  "csukuangfj/vits-piper-en_GB-southern_english_female-medium|6 speakers": _get_vits_piper,
451
+ "csukuangfj/vits-piper-en_US-bryce-medium|1 speaker": _get_vits_piper,
452
+ "csukuangfj/vits-piper-en_US-john-medium|1 speaker": _get_vits_piper,
453
+ "csukuangfj/vits-piper-en_US-norman-medium|1 speaker": _get_vits_piper,
454
  # coqui-ai
455
  "csukuangfj/vits-coqui-en-ljspeech|1 speaker": _get_vits_piper,
456
  "csukuangfj/vits-coqui-en-ljspeech-neon|1 speaker": _get_vits_piper,
 
462
  "csukuangfj/vits-piper-en_US-arctic-medium|18 speakers": _get_vits_piper, # 18 speakers
463
  "csukuangfj/vits-piper-en_US-danny-low|1 speaker": _get_vits_piper,
464
  "csukuangfj/vits-piper-en_US-hfc_male-medium|1 speaker": _get_vits_piper,
465
+ "csukuangfj/vits-piper-en_US-hfc_female-medium|1 speaker": _get_vits_piper,
466
  "csukuangfj/vits-piper-en_US-joe-medium|1 speaker": _get_vits_piper,
467
  "csukuangfj/vits-piper-en_US-kathleen-low|1 speaker": _get_vits_piper,
468
  "csukuangfj/vits-piper-en_US-kusal-medium|1 speaker": _get_vits_piper,
 
524
  "csukuangfj/vits-coqui-fr-css10": _get_vits_piper,
525
  # "csukuangfj/vits-piper-fr_FR-gilles-low": _get_vits_piper,
526
  # "csukuangfj/vits-piper-fr_FR-mls_1840-low": _get_vits_piper,
527
+ # "csukuangfj/vits-piper-fr_FR-mls-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
528
  "csukuangfj/vits-piper-fr_FR-upmc-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
529
+ "csukuangfj/vits-piper-fr_FR-tom-medium|1 speaker": _get_vits_piper, # 2 speakers, 0-femal, 1-male
530
  "csukuangfj/vits-piper-fr_FR-siwis-low": _get_vits_piper, # female
531
  "csukuangfj/vits-piper-fr_FR-siwis-medium": _get_vits_piper,
532
  "csukuangfj/vits-piper-fr_FR-tjiho-model1": _get_vits_piper,
 
598
 
599
  italian_models = {
600
  "csukuangfj/vits-piper-it_IT-riccardo-x_low": _get_vits_piper,
601
+ "csukuangfj/vits-piper-it_IT-paola-medium": _get_vits_piper,
602
  }
603
 
604
  georgian_models = {
 
679
  turkish_models = {
680
  "csukuangfj/vits-piper-tr_TR-dfki-medium": _get_vits_piper,
681
  "csukuangfj/vits-piper-tr_TR-fahrettin-medium": _get_vits_piper,
682
+ "csukuangfj/vits-piper-tr_TR-fettah-medium|1 speaker": _get_vits_piper,
683
  }
684
 
685
  vietnamese_models = {
 
759
  "csukuangfj/vits-mimic3-tn_ZA-google-nwu_low": _get_vits_piper,
760
  }
761
 
762
+ welsh_models = {
763
+ "csukuangfj/vits-piper-cy_GB-gwryw_gogleddol-medium|1 speaker": _get_vits_piper,
764
+ }
765
 
766
  language_to_models = {
767
  "English": list(english_models.keys()),
 
812
  "Turkish": list(turkish_models.keys()),
813
  "Ukrainian": list(ukrainian_models.keys()),
814
  "Vietnamese": list(vietnamese_models.keys()),
815
+ "Welsh": list(welsh_models.keys()),
816
  }