Spaces:
Running
Running
csukuangfj
commited on
Commit
•
9e48595
1
Parent(s):
7fcf32c
add more models
Browse files
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 |
}
|