csukuangfj commited on
Commit
9e48595
1 Parent(s): 7fcf32c

add more models

Browse files
Files changed (1) hide show
  1. model.py +171 -0
model.py CHANGED
@@ -275,6 +275,48 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
275
  return catalan_models[repo_id](repo_id, speed)
276
  elif repo_id in czech_models:
277
  return czech_models[repo_id](repo_id, speed)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
278
  else:
279
  raise ValueError(f"Unsupported repo_id: {repo_id}")
280
 
@@ -398,6 +440,114 @@ czech_models = {
398
  "csukuangfj/vits-piper-cs_CZ-jirka-medium": _get_vits_piper,
399
  }
400
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
401
  language_to_models = {
402
  "English": list(english_models.keys()),
403
  "Chinese (Mandarin, 普通话)": list(chinese_models.keys()),
@@ -405,9 +555,30 @@ language_to_models = {
405
  "Arabic": list(arabic_models.keys()),
406
  "Catalan": list(catalan_models.keys()),
407
  "Czech": list(czech_models.keys()),
 
 
 
408
  "French": list(french_models.keys()),
 
409
  "German": list(german_models.keys()),
 
 
 
 
 
 
 
 
 
 
 
410
  "Russian": list(russian_models.keys()),
 
 
411
  "Spanish": list(spanish_models.keys()),
 
 
 
412
  "Ukrainian": list(ukrainian_models.keys()),
 
413
  }
 
275
  return catalan_models[repo_id](repo_id, speed)
276
  elif repo_id in czech_models:
277
  return czech_models[repo_id](repo_id, speed)
278
+ elif repo_id in danish_models:
279
+ return danish_models[repo_id](repo_id, speed)
280
+ elif repo_id in greek_models:
281
+ return greek_models[repo_id](repo_id, speed)
282
+ elif repo_id in finnish_models:
283
+ return finnish_models[repo_id](repo_id, speed)
284
+ elif repo_id in hungarian_models:
285
+ return hungarian_models[repo_id](repo_id, speed)
286
+ elif repo_id in icelandic_models:
287
+ return icelandic_models[repo_id](repo_id, speed)
288
+ elif repo_id in italian_models:
289
+ return italian_models[repo_id](repo_id, speed)
290
+ elif repo_id in georgian_models:
291
+ return georgian_models[repo_id](repo_id, speed)
292
+ elif repo_id in kazakh_models:
293
+ return kazakh_models[repo_id](repo_id, speed)
294
+ elif repo_id in luxembourgish_models:
295
+ return luxembourgish_models[repo_id](repo_id, speed)
296
+ elif repo_id in nepali_models:
297
+ return nepali_models[repo_id](repo_id, speed)
298
+ elif repo_id in dutch_models:
299
+ return dutch_models[repo_id](repo_id, speed)
300
+ elif repo_id in norwegian_models:
301
+ return norwegian_models[repo_id](repo_id, speed)
302
+ elif repo_id in polish_models:
303
+ return polish_models[repo_id](repo_id, speed)
304
+ elif repo_id in portuguese_models:
305
+ return portuguese_models[repo_id](repo_id, speed)
306
+ elif repo_id in romanian_models:
307
+ return romanian_models[repo_id](repo_id, speed)
308
+ elif repo_id in slovak_models:
309
+ return slovak_models[repo_id](repo_id, speed)
310
+ elif repo_id in serbian_models:
311
+ return serbian_models[repo_id](repo_id, speed)
312
+ elif repo_id in swedish_models:
313
+ return swedish_models[repo_id](repo_id, speed)
314
+ elif repo_id in swahili_models:
315
+ return swahili_models[repo_id](repo_id, speed)
316
+ elif repo_id in turkish_models:
317
+ return turkish_models[repo_id](repo_id, speed)
318
+ elif repo_id in vietnamese_models:
319
+ return vietnamese_models[repo_id](repo_id, speed)
320
  else:
321
  raise ValueError(f"Unsupported repo_id: {repo_id}")
322
 
 
440
  "csukuangfj/vits-piper-cs_CZ-jirka-medium": _get_vits_piper,
441
  }
442
 
443
+ danish_models = {
444
+ "csukuangfj/vits-piper-da_DK-talesyntese-medium": _get_vits_piper,
445
+ }
446
+
447
+ greek_models = {
448
+ "csukuangfj/vits-piper-el_GR-rapunzelina-low": _get_vits_piper,
449
+ }
450
+
451
+ finnish_models = {
452
+ "csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
453
+ "csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
454
+ }
455
+
456
+ hungarian_models = {
457
+ "csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
458
+ "csukuangfj/vits-piper-hu_HU-berta-medium": _get_vits_piper,
459
+ "csukuangfj/vits-piper-hu_HU-imre-medium": _get_vits_piper,
460
+ }
461
+
462
+ icelandic_models = {
463
+ "csukuangfj/vits-piper-is_IS-bui-medium": _get_vits_piper,
464
+ "csukuangfj/vits-piper-is_IS-salka-medium": _get_vits_piper,
465
+ "csukuangfj/vits-piper-is_IS-steinn-medium": _get_vits_piper,
466
+ "csukuangfj/vits-piper-is_IS-ugla-medium": _get_vits_piper,
467
+ }
468
+
469
+ italian_models = {
470
+ "csukuangfj/vits-piper-it_IT-riccardo-x_low": _get_vits_piper,
471
+ }
472
+
473
+ georgian_models = {
474
+ "csukuangfj/vits-piper-ka_GE-natia-medium": _get_vits_piper,
475
+ }
476
+
477
+ kazakh_models = {
478
+ "csukuangfj/vits-piper-kk_KZ-iseke-x_low": _get_vits_piper,
479
+ "csukuangfj/vits-piper-kk_KZ-issai-high": _get_vits_piper,
480
+ "csukuangfj/vits-piper-kk_KZ-raya-x_low": _get_vits_piper,
481
+ }
482
+
483
+ luxembourgish_models = {
484
+ "csukuangfj/vits-piper-lb_LU-marylux-medium": _get_vits_piper,
485
+ }
486
+
487
+ nepali_models = {
488
+ "csukuangfj/vits-piper-ne_NP-google-medium": _get_vits_piper,
489
+ "csukuangfj/vits-piper-ne_NP-google-x_low": _get_vits_piper,
490
+ }
491
+
492
+ dutch_models = {
493
+ "csukuangfj/vits-piper-nl_BE-nathalie-medium": _get_vits_piper,
494
+ "csukuangfj/vits-piper-nl_BE-nathalie-x_low": _get_vits_piper,
495
+ "csukuangfj/vits-piper-nl_BE-rdh-medium": _get_vits_piper,
496
+ "csukuangfj/vits-piper-nl_BE-rdh-x_low": _get_vits_piper,
497
+ "csukuangfj/vits-piper-nl_NL-mls_5809-low": _get_vits_piper,
498
+ "csukuangfj/vits-piper-nl_NL-mls_7432-low": _get_vits_piper,
499
+ }
500
+
501
+ norwegian_models = {
502
+ "csukuangfj/vits-piper-no_NO-talesyntese-medium": _get_vits_piper,
503
+ }
504
+
505
+ polish_models = {
506
+ "csukuangfj/vits-piper-pl_PL-darkman-medium": _get_vits_piper,
507
+ "csukuangfj/vits-piper-pl_PL-gosia-medium": _get_vits_piper,
508
+ "csukuangfj/vits-piper-pl_PL-mc_speech-medium": _get_vits_piper,
509
+ "csukuangfj/vits-piper-pl_PL-mls_6892-low": _get_vits_piper,
510
+ }
511
+
512
+ portuguese_models = {
513
+ "csukuangfj/vits-piper-pt_BR-edresson-low": _get_vits_piper,
514
+ "csukuangfj/vits-piper-pt_BR-faber-medium": _get_vits_piper,
515
+ "csukuangfj/vits-piper-pt_PT-tugao-medium": _get_vits_piper,
516
+ }
517
+
518
+ romanian_models = {
519
+ "csukuangfj/vits-piper-ro_RO-mihai-medium": _get_vits_piper,
520
+ }
521
+
522
+
523
+ slovak_models = {
524
+ "csukuangfj/vits-piper-sk_SK-lili-medium": _get_vits_piper,
525
+ }
526
+
527
+ serbian_models = {
528
+ "csukuangfj/vits-piper-sr_RS-serbski_institut-medium": _get_vits_piper,
529
+ }
530
+
531
+ swedish_models = {
532
+ "csukuangfj/vits-piper-sv_SE-nst-medium": _get_vits_piper,
533
+ }
534
+
535
+ swahili_models = {
536
+ "csukuangfj/vits-piper-sw_CD-lanfrica-medium": _get_vits_piper,
537
+ }
538
+
539
+ turkish_models = {
540
+ "csukuangfj/vits-piper-tr_TR-dfki-medium": _get_vits_piper,
541
+ "csukuangfj/vits-piper-tr_TR-fahrettin-medium": _get_vits_piper,
542
+ }
543
+
544
+ vietnamese_models = {
545
+ "csukuangfj/vits-piper-vi_VN-25hours_single-low": _get_vits_piper,
546
+ "csukuangfj/vits-piper-vi_VN-vais1000-medium": _get_vits_piper,
547
+ "csukuangfj/vits-piper-vi_VN-vivos-x_low": _get_vits_piper,
548
+ }
549
+
550
+
551
  language_to_models = {
552
  "English": list(english_models.keys()),
553
  "Chinese (Mandarin, 普通话)": list(chinese_models.keys()),
 
555
  "Arabic": list(arabic_models.keys()),
556
  "Catalan": list(catalan_models.keys()),
557
  "Czech": list(czech_models.keys()),
558
+ "Danish": list(danish_models.keys()),
559
+ "Dutch": list(dutch_models.keys()),
560
+ "Finnish": list(finnish_models.keys()),
561
  "French": list(french_models.keys()),
562
+ "Georgian": list(georgian_models.keys()),
563
  "German": list(german_models.keys()),
564
+ "Greek": list(greek_models.keys()),
565
+ "Hungarian": list(hungarian_models.keys()),
566
+ "Icelandic": list(icelandic_models.keys()),
567
+ "Italian": list(italian_models.keys()),
568
+ "Kazakh": list(kazakh_models.keys()),
569
+ "Luxembourgish": list(luxembourgish_models.keys()),
570
+ "Nepali": list(nepali_models.keys()),
571
+ "Norwegian": list(norwegian_models.keys()),
572
+ "Polish": list(polish_models.keys()),
573
+ "Portuguese": list(portuguese_models.keys()),
574
+ "Romanian": list(romanian_models.keys()),
575
  "Russian": list(russian_models.keys()),
576
+ "Serbian": list(serbian_models.keys()),
577
+ "Slovak": list(slovak_models.keys()),
578
  "Spanish": list(spanish_models.keys()),
579
+ "Swahili": list(swahili_models.keys()),
580
+ "Swedish": list(swedish_models.keys()),
581
+ "Turkish": list(turkish_models.keys()),
582
  "Ukrainian": list(ukrainian_models.keys()),
583
+ "Vietnamese": list(vietnamese_models.keys()),
584
  }