Spaces:
Runtime error
Runtime error
Commit
·
1ae3349
1
Parent(s):
0ea07f8
add coqui models
Browse files
model.py
CHANGED
|
@@ -319,6 +319,26 @@ def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
|
|
| 319 |
return turkish_models[repo_id](repo_id, speed)
|
| 320 |
elif repo_id in vietnamese_models:
|
| 321 |
return vietnamese_models[repo_id](repo_id, speed)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 322 |
else:
|
| 323 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
| 324 |
|
|
@@ -385,6 +405,7 @@ english_models = {
|
|
| 385 |
}
|
| 386 |
|
| 387 |
german_models = {
|
|
|
|
| 388 |
"csukuangfj/vits-piper-de_DE-eva_k-x_low": _get_vits_piper,
|
| 389 |
"csukuangfj/vits-piper-de_DE-karlsson-low": _get_vits_piper,
|
| 390 |
"csukuangfj/vits-piper-de_DE-kerstin-low": _get_vits_piper,
|
|
@@ -397,6 +418,7 @@ german_models = {
|
|
| 397 |
}
|
| 398 |
|
| 399 |
spanish_models = {
|
|
|
|
| 400 |
"csukuangfj/vits-piper-es_ES-carlfm-x_low": _get_vits_piper,
|
| 401 |
"csukuangfj/vits-piper-es_ES-davefx-medium": _get_vits_piper,
|
| 402 |
"csukuangfj/vits-piper-es_ES-mls_10246-low": _get_vits_piper,
|
|
@@ -406,6 +428,7 @@ spanish_models = {
|
|
| 406 |
}
|
| 407 |
|
| 408 |
french_models = {
|
|
|
|
| 409 |
# "csukuangfj/vits-piper-fr_FR-gilles-low": _get_vits_piper,
|
| 410 |
# "csukuangfj/vits-piper-fr_FR-mls_1840-low": _get_vits_piper,
|
| 411 |
"csukuangfj/vits-piper-fr_FR-upmc-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
|
|
@@ -443,6 +466,7 @@ catalan_models = {
|
|
| 443 |
czech_models = {
|
| 444 |
"csukuangfj/vits-piper-cs_CZ-jirka-low": _get_vits_piper,
|
| 445 |
"csukuangfj/vits-piper-cs_CZ-jirka-medium": _get_vits_piper,
|
|
|
|
| 446 |
}
|
| 447 |
|
| 448 |
danish_models = {
|
|
@@ -454,11 +478,13 @@ greek_models = {
|
|
| 454 |
}
|
| 455 |
|
| 456 |
finnish_models = {
|
|
|
|
| 457 |
"csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
|
| 458 |
"csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
|
| 459 |
}
|
| 460 |
|
| 461 |
hungarian_models = {
|
|
|
|
| 462 |
"csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
|
| 463 |
"csukuangfj/vits-piper-hu_HU-berta-medium": _get_vits_piper,
|
| 464 |
"csukuangfj/vits-piper-hu_HU-imre-medium": _get_vits_piper,
|
|
@@ -495,6 +521,7 @@ nepali_models = {
|
|
| 495 |
}
|
| 496 |
|
| 497 |
dutch_models = {
|
|
|
|
| 498 |
"csukuangfj/vits-piper-nl_BE-nathalie-medium": _get_vits_piper,
|
| 499 |
"csukuangfj/vits-piper-nl_BE-nathalie-x_low": _get_vits_piper,
|
| 500 |
"csukuangfj/vits-piper-nl_BE-rdh-medium": _get_vits_piper,
|
|
@@ -508,6 +535,7 @@ norwegian_models = {
|
|
| 508 |
}
|
| 509 |
|
| 510 |
polish_models = {
|
|
|
|
| 511 |
"csukuangfj/vits-piper-pl_PL-darkman-medium": _get_vits_piper,
|
| 512 |
"csukuangfj/vits-piper-pl_PL-gosia-medium": _get_vits_piper,
|
| 513 |
"csukuangfj/vits-piper-pl_PL-mc_speech-medium": _get_vits_piper,
|
|
@@ -515,17 +543,20 @@ polish_models = {
|
|
| 515 |
}
|
| 516 |
|
| 517 |
portuguese_models = {
|
|
|
|
| 518 |
"csukuangfj/vits-piper-pt_BR-edresson-low": _get_vits_piper,
|
| 519 |
"csukuangfj/vits-piper-pt_BR-faber-medium": _get_vits_piper,
|
| 520 |
"csukuangfj/vits-piper-pt_PT-tugao-medium": _get_vits_piper,
|
| 521 |
}
|
| 522 |
|
| 523 |
romanian_models = {
|
|
|
|
| 524 |
"csukuangfj/vits-piper-ro_RO-mihai-medium": _get_vits_piper,
|
| 525 |
}
|
| 526 |
|
| 527 |
|
| 528 |
slovak_models = {
|
|
|
|
| 529 |
"csukuangfj/vits-piper-sk_SK-lili-medium": _get_vits_piper,
|
| 530 |
}
|
| 531 |
|
|
@@ -534,6 +565,7 @@ serbian_models = {
|
|
| 534 |
}
|
| 535 |
|
| 536 |
swedish_models = {
|
|
|
|
| 537 |
"csukuangfj/vits-piper-sv_SE-nst-medium": _get_vits_piper,
|
| 538 |
}
|
| 539 |
|
|
@@ -552,16 +584,61 @@ vietnamese_models = {
|
|
| 552 |
"csukuangfj/vits-piper-vi_VN-vivos-x_low": _get_vits_piper,
|
| 553 |
}
|
| 554 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 555 |
|
| 556 |
language_to_models = {
|
| 557 |
"English": list(english_models.keys()),
|
| 558 |
"Chinese (Mandarin, 普通话)": list(chinese_models.keys()),
|
| 559 |
"Cantonese (粤语)": list(cantonese_models.keys()),
|
| 560 |
"Arabic": list(arabic_models.keys()),
|
|
|
|
|
|
|
| 561 |
"Catalan": list(catalan_models.keys()),
|
|
|
|
| 562 |
"Czech": list(czech_models.keys()),
|
| 563 |
"Danish": list(danish_models.keys()),
|
|
|
|
| 564 |
"Dutch": list(dutch_models.keys()),
|
|
|
|
| 565 |
"Finnish": list(finnish_models.keys()),
|
| 566 |
"French": list(french_models.keys()),
|
| 567 |
"Georgian": list(georgian_models.keys()),
|
|
@@ -569,9 +646,13 @@ language_to_models = {
|
|
| 569 |
"Greek": list(greek_models.keys()),
|
| 570 |
"Hungarian": list(hungarian_models.keys()),
|
| 571 |
"Icelandic": list(icelandic_models.keys()),
|
|
|
|
| 572 |
"Italian": list(italian_models.keys()),
|
| 573 |
"Kazakh": list(kazakh_models.keys()),
|
|
|
|
|
|
|
| 574 |
"Luxembourgish": list(luxembourgish_models.keys()),
|
|
|
|
| 575 |
"Nepali": list(nepali_models.keys()),
|
| 576 |
"Norwegian": list(norwegian_models.keys()),
|
| 577 |
"Polish": list(polish_models.keys()),
|
|
@@ -580,6 +661,7 @@ language_to_models = {
|
|
| 580 |
"Russian": list(russian_models.keys()),
|
| 581 |
"Serbian": list(serbian_models.keys()),
|
| 582 |
"Slovak": list(slovak_models.keys()),
|
|
|
|
| 583 |
"Spanish": list(spanish_models.keys()),
|
| 584 |
"Swahili": list(swahili_models.keys()),
|
| 585 |
"Swedish": list(swedish_models.keys()),
|
|
|
|
| 319 |
return turkish_models[repo_id](repo_id, speed)
|
| 320 |
elif repo_id in vietnamese_models:
|
| 321 |
return vietnamese_models[repo_id](repo_id, speed)
|
| 322 |
+
elif repo_id in bulgarian_models:
|
| 323 |
+
return bulgarian_models[repo_id](repo_id, speed)
|
| 324 |
+
elif repo_id in danish_models:
|
| 325 |
+
return danish_models[repo_id](repo_id, speed)
|
| 326 |
+
elif repo_id in estonian_models:
|
| 327 |
+
return estonian_models[repo_id](repo_id, speed)
|
| 328 |
+
elif repo_id in irish_models:
|
| 329 |
+
return irish_models[repo_id](repo_id, speed)
|
| 330 |
+
elif repo_id in croatian_models:
|
| 331 |
+
return croatian_models[repo_id](repo_id, speed)
|
| 332 |
+
elif repo_id in lithuanian_models:
|
| 333 |
+
return lithuanian_models[repo_id](repo_id, speed)
|
| 334 |
+
elif repo_id in latvian_models:
|
| 335 |
+
return lithuanian_models[repo_id](repo_id, speed)
|
| 336 |
+
elif repo_id in maltese_models:
|
| 337 |
+
return maltese_models[repo_id](repo_id, speed)
|
| 338 |
+
elif repo_id in slovenian_models:
|
| 339 |
+
return slovenian_models[repo_id](repo_id, speed)
|
| 340 |
+
elif repo_id in bangla_models:
|
| 341 |
+
return bangla_models[repo_id](repo_id, speed)
|
| 342 |
else:
|
| 343 |
raise ValueError(f"Unsupported repo_id: {repo_id}")
|
| 344 |
|
|
|
|
| 405 |
}
|
| 406 |
|
| 407 |
german_models = {
|
| 408 |
+
"csukuangfj/vits-coqui-de-css10": _get_vits_piper,
|
| 409 |
"csukuangfj/vits-piper-de_DE-eva_k-x_low": _get_vits_piper,
|
| 410 |
"csukuangfj/vits-piper-de_DE-karlsson-low": _get_vits_piper,
|
| 411 |
"csukuangfj/vits-piper-de_DE-kerstin-low": _get_vits_piper,
|
|
|
|
| 418 |
}
|
| 419 |
|
| 420 |
spanish_models = {
|
| 421 |
+
"csukuangfj/vits-coqui-es-css10": _get_vits_piper,
|
| 422 |
"csukuangfj/vits-piper-es_ES-carlfm-x_low": _get_vits_piper,
|
| 423 |
"csukuangfj/vits-piper-es_ES-davefx-medium": _get_vits_piper,
|
| 424 |
"csukuangfj/vits-piper-es_ES-mls_10246-low": _get_vits_piper,
|
|
|
|
| 428 |
}
|
| 429 |
|
| 430 |
french_models = {
|
| 431 |
+
"csukuangfj/vits-coqui-fr-css10": _get_vits_piper,
|
| 432 |
# "csukuangfj/vits-piper-fr_FR-gilles-low": _get_vits_piper,
|
| 433 |
# "csukuangfj/vits-piper-fr_FR-mls_1840-low": _get_vits_piper,
|
| 434 |
"csukuangfj/vits-piper-fr_FR-upmc-medium": _get_vits_piper, # 2 speakers, 0-femal, 1-male
|
|
|
|
| 466 |
czech_models = {
|
| 467 |
"csukuangfj/vits-piper-cs_CZ-jirka-low": _get_vits_piper,
|
| 468 |
"csukuangfj/vits-piper-cs_CZ-jirka-medium": _get_vits_piper,
|
| 469 |
+
"csukuangfj/vits-coqui-cs-cv": _get_vits_piper,
|
| 470 |
}
|
| 471 |
|
| 472 |
danish_models = {
|
|
|
|
| 478 |
}
|
| 479 |
|
| 480 |
finnish_models = {
|
| 481 |
+
"csukuangfj/vits-coqui-fi-css10": _get_vits_piper,
|
| 482 |
"csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
|
| 483 |
"csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
|
| 484 |
}
|
| 485 |
|
| 486 |
hungarian_models = {
|
| 487 |
+
"csukuangfj/vits-coqui-hu-css10": _get_vits_piper,
|
| 488 |
"csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
|
| 489 |
"csukuangfj/vits-piper-hu_HU-berta-medium": _get_vits_piper,
|
| 490 |
"csukuangfj/vits-piper-hu_HU-imre-medium": _get_vits_piper,
|
|
|
|
| 521 |
}
|
| 522 |
|
| 523 |
dutch_models = {
|
| 524 |
+
"csukuangfj/vits-coqui-nl-css10": _get_vits_piper,
|
| 525 |
"csukuangfj/vits-piper-nl_BE-nathalie-medium": _get_vits_piper,
|
| 526 |
"csukuangfj/vits-piper-nl_BE-nathalie-x_low": _get_vits_piper,
|
| 527 |
"csukuangfj/vits-piper-nl_BE-rdh-medium": _get_vits_piper,
|
|
|
|
| 535 |
}
|
| 536 |
|
| 537 |
polish_models = {
|
| 538 |
+
"csukuangfj/vits-coqui-pl-mai_female": _get_vits_piper,
|
| 539 |
"csukuangfj/vits-piper-pl_PL-darkman-medium": _get_vits_piper,
|
| 540 |
"csukuangfj/vits-piper-pl_PL-gosia-medium": _get_vits_piper,
|
| 541 |
"csukuangfj/vits-piper-pl_PL-mc_speech-medium": _get_vits_piper,
|
|
|
|
| 543 |
}
|
| 544 |
|
| 545 |
portuguese_models = {
|
| 546 |
+
"csukuangfj/vits-coqui-pt-cv": _get_vits_piper,
|
| 547 |
"csukuangfj/vits-piper-pt_BR-edresson-low": _get_vits_piper,
|
| 548 |
"csukuangfj/vits-piper-pt_BR-faber-medium": _get_vits_piper,
|
| 549 |
"csukuangfj/vits-piper-pt_PT-tugao-medium": _get_vits_piper,
|
| 550 |
}
|
| 551 |
|
| 552 |
romanian_models = {
|
| 553 |
+
"csukuangfj/vits-coqui-ro-cv": _get_vits_piper,
|
| 554 |
"csukuangfj/vits-piper-ro_RO-mihai-medium": _get_vits_piper,
|
| 555 |
}
|
| 556 |
|
| 557 |
|
| 558 |
slovak_models = {
|
| 559 |
+
"csukuangfj/vits-coqui-sk-cv": _get_vits_piper,
|
| 560 |
"csukuangfj/vits-piper-sk_SK-lili-medium": _get_vits_piper,
|
| 561 |
}
|
| 562 |
|
|
|
|
| 565 |
}
|
| 566 |
|
| 567 |
swedish_models = {
|
| 568 |
+
"csukuangfj/vits-coqui-sv-cv": _get_vits_piper,
|
| 569 |
"csukuangfj/vits-piper-sv_SE-nst-medium": _get_vits_piper,
|
| 570 |
}
|
| 571 |
|
|
|
|
| 584 |
"csukuangfj/vits-piper-vi_VN-vivos-x_low": _get_vits_piper,
|
| 585 |
}
|
| 586 |
|
| 587 |
+
bulgarian_models = {
|
| 588 |
+
"csukuangfj/vits-coqui-bg-cv": _get_vits_piper,
|
| 589 |
+
}
|
| 590 |
+
|
| 591 |
+
danish_models = {
|
| 592 |
+
"csukuangfj/vits-coqui-da-cv": _get_vits_piper,
|
| 593 |
+
}
|
| 594 |
+
|
| 595 |
+
estonian_models = {
|
| 596 |
+
"csukuangfj/vits-coqui-et-cv": _get_vits_piper,
|
| 597 |
+
}
|
| 598 |
+
|
| 599 |
+
irish_models = {
|
| 600 |
+
"csukuangfj/vits-coqui-ga-cv": _get_vits_piper,
|
| 601 |
+
}
|
| 602 |
+
|
| 603 |
+
croatian_models = {
|
| 604 |
+
"csukuangfj/vits-coqui-hr-cv": _get_vits_piper,
|
| 605 |
+
}
|
| 606 |
+
|
| 607 |
+
lithuanian_models = {
|
| 608 |
+
"csukuangfj/vits-coqui-lt-cv": _get_vits_piper,
|
| 609 |
+
}
|
| 610 |
+
|
| 611 |
+
latvian_models = {
|
| 612 |
+
"csukuangfj/vits-coqui-lv-cv": _get_vits_piper,
|
| 613 |
+
}
|
| 614 |
+
|
| 615 |
+
maltese_models = {
|
| 616 |
+
"csukuangfj/vits-coqui-mt-cv": _get_vits_piper,
|
| 617 |
+
}
|
| 618 |
+
|
| 619 |
+
slovenian_models = {
|
| 620 |
+
"csukuangfj/vits-coqui-sl-cv": _get_vits_piper,
|
| 621 |
+
}
|
| 622 |
+
|
| 623 |
+
bangla_models = {
|
| 624 |
+
"csukuangfj/vits-coqui-sl-cv": _get_vits_piper,
|
| 625 |
+
}
|
| 626 |
+
|
| 627 |
|
| 628 |
language_to_models = {
|
| 629 |
"English": list(english_models.keys()),
|
| 630 |
"Chinese (Mandarin, 普通话)": list(chinese_models.keys()),
|
| 631 |
"Cantonese (粤语)": list(cantonese_models.keys()),
|
| 632 |
"Arabic": list(arabic_models.keys()),
|
| 633 |
+
"Bangla": list(bangla_models.keys()),
|
| 634 |
+
"Bulgarian": list(Bulgarian_models.keys()),
|
| 635 |
"Catalan": list(catalan_models.keys()),
|
| 636 |
+
"Croatian": list(croatian_models.keys()),
|
| 637 |
"Czech": list(czech_models.keys()),
|
| 638 |
"Danish": list(danish_models.keys()),
|
| 639 |
+
"Danish": list(danish_models.keys()),
|
| 640 |
"Dutch": list(dutch_models.keys()),
|
| 641 |
+
"Estonian": list(estonian_models.keys()),
|
| 642 |
"Finnish": list(finnish_models.keys()),
|
| 643 |
"French": list(french_models.keys()),
|
| 644 |
"Georgian": list(georgian_models.keys()),
|
|
|
|
| 646 |
"Greek": list(greek_models.keys()),
|
| 647 |
"Hungarian": list(hungarian_models.keys()),
|
| 648 |
"Icelandic": list(icelandic_models.keys()),
|
| 649 |
+
"Irish": list(irish_models.keys()),
|
| 650 |
"Italian": list(italian_models.keys()),
|
| 651 |
"Kazakh": list(kazakh_models.keys()),
|
| 652 |
+
"Latvian": list(latvian_models.keys()),
|
| 653 |
+
"Lithuanian": list(lithuanian_models.keys()),
|
| 654 |
"Luxembourgish": list(luxembourgish_models.keys()),
|
| 655 |
+
"Maltese": list(maltese_models.keys()),
|
| 656 |
"Nepali": list(nepali_models.keys()),
|
| 657 |
"Norwegian": list(norwegian_models.keys()),
|
| 658 |
"Polish": list(polish_models.keys()),
|
|
|
|
| 661 |
"Russian": list(russian_models.keys()),
|
| 662 |
"Serbian": list(serbian_models.keys()),
|
| 663 |
"Slovak": list(slovak_models.keys()),
|
| 664 |
+
"Slovenian": list(slovenian_models.keys()),
|
| 665 |
"Spanish": list(spanish_models.keys()),
|
| 666 |
"Swahili": list(swahili_models.keys()),
|
| 667 |
"Swedish": list(swedish_models.keys()),
|