removed all languajes but spanish

model.py

@@ -265,428 +265,15 @@ def _get_vits_hf(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
 
 @lru_cache(maxsize=10)
 def get_pretrained_model(repo_id: str, speed: float) -> sherpa_onnx.OfflineTts:
-    if repo_id in chinese_models:
-        return chinese_models[repo_id](repo_id, speed)
-    if repo_id in cantonese_models:
-        return cantonese_models[repo_id](repo_id, speed)
-    elif repo_id in english_models:
-        return english_models[repo_id](repo_id, speed)
-    elif repo_id in german_models:
-        return german_models[repo_id](repo_id, speed)
-    elif repo_id in spanish_models:
+    if repo_id in spanish_models:
         return spanish_models[repo_id](repo_id, speed)
-    elif repo_id in french_models:
-        return french_models[repo_id](repo_id, speed)
-    elif repo_id in ukrainian_models:
-        return ukrainian_models[repo_id](repo_id, speed)
-    elif repo_id in russian_models:
-        return russian_models[repo_id](repo_id, speed)
-    elif repo_id in arabic_models:
-        return arabic_models[repo_id](repo_id, speed)
-    elif repo_id in catalan_models:
-        return catalan_models[repo_id](repo_id, speed)
-    elif repo_id in czech_models:
-        return czech_models[repo_id](repo_id, speed)
-    elif repo_id in danish_models:
-        return danish_models[repo_id](repo_id, speed)
-    elif repo_id in greek_models:
-        return greek_models[repo_id](repo_id, speed)
-    elif repo_id in finnish_models:
-        return finnish_models[repo_id](repo_id, speed)
-    elif repo_id in hungarian_models:
-        return hungarian_models[repo_id](repo_id, speed)
-    elif repo_id in icelandic_models:
-        return icelandic_models[repo_id](repo_id, speed)
-    elif repo_id in italian_models:
-        return italian_models[repo_id](repo_id, speed)
-    elif repo_id in georgian_models:
-        return georgian_models[repo_id](repo_id, speed)
-    elif repo_id in kazakh_models:
-        return kazakh_models[repo_id](repo_id, speed)
-    elif repo_id in luxembourgish_models:
-        return luxembourgish_models[repo_id](repo_id, speed)
-    elif repo_id in nepali_models:
-        return nepali_models[repo_id](repo_id, speed)
-    elif repo_id in dutch_models:
-        return dutch_models[repo_id](repo_id, speed)
-    elif repo_id in norwegian_models:
-        return norwegian_models[repo_id](repo_id, speed)
-    elif repo_id in polish_models:
-        return polish_models[repo_id](repo_id, speed)
-    elif repo_id in portuguese_models:
-        return portuguese_models[repo_id](repo_id, speed)
-    elif repo_id in romanian_models:
-        return romanian_models[repo_id](repo_id, speed)
-    elif repo_id in slovak_models:
-        return slovak_models[repo_id](repo_id, speed)
-    elif repo_id in serbian_models:
-        return serbian_models[repo_id](repo_id, speed)
-    elif repo_id in swedish_models:
-        return swedish_models[repo_id](repo_id, speed)
-    elif repo_id in swahili_models:
-        return swahili_models[repo_id](repo_id, speed)
-    elif repo_id in turkish_models:
-        return turkish_models[repo_id](repo_id, speed)
-    elif repo_id in vietnamese_models:
-        return vietnamese_models[repo_id](repo_id, speed)
-    elif repo_id in bulgarian_models:
-        return bulgarian_models[repo_id](repo_id, speed)
-    elif repo_id in estonian_models:
-        return estonian_models[repo_id](repo_id, speed)
-    elif repo_id in irish_models:
-        return irish_models[repo_id](repo_id, speed)
-    elif repo_id in croatian_models:
-        return croatian_models[repo_id](repo_id, speed)
-    elif repo_id in lithuanian_models:
-        return lithuanian_models[repo_id](repo_id, speed)
-    elif repo_id in latvian_models:
-        return latvian_models[repo_id](repo_id, speed)
-    elif repo_id in maltese_models:
-        return maltese_models[repo_id](repo_id, speed)
-    elif repo_id in slovenian_models:
-        return slovenian_models[repo_id](repo_id, speed)
-    elif repo_id in bengali_models:
-        return bengali_models[repo_id](repo_id, speed)
-    elif repo_id in min_nan_models:
-        return min_nan_models[repo_id](repo_id, speed)
-    elif repo_id in thai_models:
-        return thai_models[repo_id](repo_id, speed)
-    elif repo_id in persian_models:
-        return persian_models[repo_id](repo_id, speed)
-    elif repo_id in korean_models:
-        return korean_models[repo_id](repo_id, speed)
-    elif repo_id in afrikaans_models:
-        return afrikaans_models[repo_id](repo_id, speed)
-    elif repo_id in gujarati_models:
-        return gujarati_models[repo_id](repo_id, speed)
-    elif repo_id in tswana_models:
-        return tswana_models[repo_id](repo_id, speed)
     else:
         raise ValueError(f"Unsupported repo_id: {repo_id}")
 
-
-cantonese_models = {
-    "csukuangfj/vits-cantonese-hf-xiaomaiiwn": _get_vits_hf,
-}
-
-chinese_models = {
-    "csukuangfj/vits-piper-zh_CN-huayan-medium": _get_vits_piper,
-    "csukuangfj/vits-zh-hf-theresa": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-eula": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-echo": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-bronya": _get_vits_hf,
-    "csukuangfj/vits-zh-aishell3": _get_vits_zh_aishell3,
-    "csukuangfj/vits-zh-hf-fanchen-wnj": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-fanchen-C": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-fanchen-ZhiHuiLaoZhe": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-fanchen-ZhiHuiLaoZhe_new": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-fanchen-unity": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-doom": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-zenyatta": _get_vits_hf,  # 804
-    "csukuangfj/vits-zh-hf-abyssinvoker": _get_vits_hf,
-    "csukuangfj/vits-zh-hf-keqing": _get_vits_hf,
-    #  "csukuangfj/vits-piper-zh_CN-huayan-x_low": _get_vits_piper,
-}
-
-english_models = {
-    "csukuangfj/vits-piper-en_US-glados": _get_vits_piper,
-    # coqui-ai
-    "csukuangfj/vits-coqui-en-ljspeech": _get_vits_piper,
-    "csukuangfj/vits-coqui-en-ljspeech-neon": _get_vits_piper,
-    "csukuangfj/vits-coqui-en-vctk": _get_vits_piper,
-    # piper, US
-    "csukuangfj/vits-piper-en_GB-sweetbbak-amy": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-amy-low": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-amy-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-arctic-medium": _get_vits_piper,  # 18 speakers
-    "csukuangfj/vits-piper-en_US-danny-low": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-hfc_male-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-joe-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-kathleen-low": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-kusal-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-l2arctic-medium": _get_vits_piper,  # 24 speakers
-    "csukuangfj/vits-piper-en_US-lessac-low": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-lessac-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-lessac-high": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-libritts-high": _get_vits_piper,  # 904 speakers
-    "csukuangfj/vits-piper-en_US-libritts_r-medium": _get_vits_piper,  # 904 speakers
-    "csukuangfj/vits-piper-en_US-ryan-low": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-ryan-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_US-ryan-high": _get_vits_piper,
-    # piper, GB
-    "csukuangfj/vits-piper-en_GB-alan-low": _get_vits_piper,
-    "csukuangfj/vits-piper-en_GB-alan-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_GB-alba-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_GB-jenny_dioco-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_GB-northern_english_male-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_GB-semaine-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-en_GB-southern_english_female-low": _get_vits_piper,
-    "csukuangfj/vits-piper-en_GB-vctk-medium": _get_vits_piper,
-    #
-    "csukuangfj/vits-vctk": _get_vits_vctk,  # 109 speakers
-    "csukuangfj/vits-ljs": _get_vits_ljs,
-}
-
-german_models = {
-    "csukuangfj/vits-coqui-de-css10": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-eva_k-x_low": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-karlsson-low": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-kerstin-low": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-pavoque-low": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-ramona-low": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-thorsten-low": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-thorsten-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-thorsten-high": _get_vits_piper,
-    "csukuangfj/vits-piper-de_DE-thorsten_emotional-medium": _get_vits_piper,  # 8 speakers
-}
-
 spanish_models = {
-    "csukuangfj/vits-coqui-es-css10": _get_vits_piper,
-    "csukuangfj/vits-piper-es-glados-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-es_ES-carlfm-x_low": _get_vits_piper,
-    "csukuangfj/vits-piper-es_ES-davefx-medium": _get_vits_piper,
-    #  "csukuangfj/vits-piper-es_ES-mls_10246-low": _get_vits_piper,
-    #  "csukuangfj/vits-piper-es_ES-mls_9972-low": _get_vits_piper,
     "csukuangfj/vits-piper-es_ES-sharvard-medium": _get_vits_piper,  # 2 speakers
-    "csukuangfj/vits-piper-es_MX-ald-medium": _get_vits_piper,
-    "csukuangfj/vits-mimic3-es_ES-m-ailabs_low": _get_vits_piper,
-}
-
-french_models = {
-    "csukuangfj/vits-coqui-fr-css10": _get_vits_piper,
-    #  "csukuangfj/vits-piper-fr_FR-gilles-low": _get_vits_piper,
-    #  "csukuangfj/vits-piper-fr_FR-mls_1840-low": _get_vits_piper,
-    "csukuangfj/vits-piper-fr_FR-upmc-medium": _get_vits_piper,  # 2 speakers, 0-femal, 1-male
-    "csukuangfj/vits-piper-fr_FR-siwis-low": _get_vits_piper,  # female
-    "csukuangfj/vits-piper-fr_FR-siwis-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-fr_FR-tjiho-model1": _get_vits_piper,
-    "csukuangfj/vits-piper-fr_FR-tjiho-model2": _get_vits_piper,
-    "csukuangfj/vits-piper-fr_FR-tjiho-model3": _get_vits_piper,
-}
-
-ukrainian_models = {
-    "csukuangfj/vits-piper-uk_UA-lada-x_low": _get_vits_piper,
-    "csukuangfj/vits-coqui-uk-mai": _get_vits_piper,
-    #  "csukuangfj/vits-piper-uk_UA-ukrainian_tts-medium": _get_vits_piper, # does not work somehow
-}
-
-russian_models = {
-    "csukuangfj/vits-piper-ru_RU-denis-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-ru_RU-dmitri-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-ru_RU-irina-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-ru_RU-ruslan-medium": _get_vits_piper,
-}
-
-arabic_models = {
-    "csukuangfj/vits-piper-ar_JO-kareem-low": _get_vits_piper,
-    "csukuangfj/vits-piper-ar_JO-kareem-medium": _get_vits_piper,
-}
-
-catalan_models = {
-    "csukuangfj/vits-piper-ca_ES-upc_ona-x_low": _get_vits_piper,
-    "csukuangfj/vits-piper-ca_ES-upc_ona-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-ca_ES-upc_pau-x_low": _get_vits_piper,
-}
-
-czech_models = {
-    "csukuangfj/vits-piper-cs_CZ-jirka-low": _get_vits_piper,
-    "csukuangfj/vits-piper-cs_CZ-jirka-medium": _get_vits_piper,
-    "csukuangfj/vits-coqui-cs-cv": _get_vits_piper,
-}
-
-danish_models = {
-    "csukuangfj/vits-coqui-da-cv": _get_vits_piper,
-    "csukuangfj/vits-piper-da_DK-talesyntese-medium": _get_vits_piper,
-}
-
-greek_models = {
-    "csukuangfj/vits-piper-el_GR-rapunzelina-low": _get_vits_piper,
-    #  "csukuangfj/vits-mimic3-el_GR-rapunzelina_low": _get_vits_piper,
-}
-
-finnish_models = {
-    "csukuangfj/vits-coqui-fi-css10": _get_vits_piper,
-    "csukuangfj/vits-piper-fi_FI-harri-low": _get_vits_piper,
-    "csukuangfj/vits-piper-fi_FI-harri-medium": _get_vits_piper,
-    "csukuangfj/vits-mimic3-fi_FI-harri-tapani-ylilammi_low": _get_vits_piper,
-}
-
-hungarian_models = {
-    #  "csukuangfj/vits-coqui-hu-css10": _get_vits_piper,
-    "csukuangfj/vits-piper-hu_HU-anna-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-hu_HU-berta-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-hu_HU-imre-medium": _get_vits_piper,
-    "csukuangfj/vits-mimic3-hu_HU-diana-majlinger_low": _get_vits_piper,
-}
-
-icelandic_models = {
-    "csukuangfj/vits-piper-is_IS-bui-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-is_IS-salka-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-is_IS-steinn-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-is_IS-ugla-medium": _get_vits_piper,
-}
-
-italian_models = {
-    "csukuangfj/vits-piper-it_IT-riccardo-x_low": _get_vits_piper,
-}
-
-georgian_models = {
-    "csukuangfj/vits-piper-ka_GE-natia-medium": _get_vits_piper,
-}
-
-kazakh_models = {
-    "csukuangfj/vits-piper-kk_KZ-iseke-x_low": _get_vits_piper,
-    "csukuangfj/vits-piper-kk_KZ-issai-high": _get_vits_piper,
-    "csukuangfj/vits-piper-kk_KZ-raya-x_low": _get_vits_piper,
-}
-
-luxembourgish_models = {
-    "csukuangfj/vits-piper-lb_LU-marylux-medium": _get_vits_piper,
-}
-
-nepali_models = {
-    "csukuangfj/vits-piper-ne_NP-google-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-ne_NP-google-x_low": _get_vits_piper,
-    "csukuangfj/vits-mimic3-ne_NP-ne-google_low": _get_vits_piper,
-}
-
-dutch_models = {
-    "csukuangfj/vits-coqui-nl-css10": _get_vits_piper,
-    "csukuangfj/vits-piper-nl_BE-nathalie-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-nl_BE-nathalie-x_low": _get_vits_piper,
-    "csukuangfj/vits-piper-nl_BE-rdh-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-nl_BE-rdh-x_low": _get_vits_piper,
-    "csukuangfj/vits-piper-nl_NL-mls_5809-low": _get_vits_piper,
-    "csukuangfj/vits-piper-nl_NL-mls_7432-low": _get_vits_piper,
-}
-
-norwegian_models = {
-    "csukuangfj/vits-piper-no_NO-talesyntese-medium": _get_vits_piper,
-}
-
-polish_models = {
-    "csukuangfj/vits-coqui-pl-mai_female": _get_vits_piper,
-    "csukuangfj/vits-piper-pl_PL-darkman-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-pl_PL-gosia-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-pl_PL-mc_speech-medium": _get_vits_piper,
-    #  "csukuangfj/vits-piper-pl_PL-mls_6892-low": _get_vits_piper,
-    "csukuangfj/vits-mimic3-pl_PL-m-ailabs_low": _get_vits_piper,
-}
-
-portuguese_models = {
-    "csukuangfj/vits-coqui-pt-cv": _get_vits_piper,
-    "csukuangfj/vits-piper-pt_BR-edresson-low": _get_vits_piper,
-    "csukuangfj/vits-piper-pt_BR-faber-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-pt_PT-tugao-medium": _get_vits_piper,
 }
 
-romanian_models = {
-    "csukuangfj/vits-coqui-ro-cv": _get_vits_piper,
-    "csukuangfj/vits-piper-ro_RO-mihai-medium": _get_vits_piper,
-}
-
-
-slovak_models = {
-    "csukuangfj/vits-coqui-sk-cv": _get_vits_piper,
-    "csukuangfj/vits-piper-sk_SK-lili-medium": _get_vits_piper,
-}
-
-serbian_models = {
-    "csukuangfj/vits-piper-sr_RS-serbski_institut-medium": _get_vits_piper,
-}
-
-swedish_models = {
-    "csukuangfj/vits-coqui-sv-cv": _get_vits_piper,
-    "csukuangfj/vits-piper-sv_SE-nst-medium": _get_vits_piper,
-}
-
-swahili_models = {
-    "csukuangfj/vits-piper-sw_CD-lanfrica-medium": _get_vits_piper,
-}
-
-turkish_models = {
-    "csukuangfj/vits-piper-tr_TR-dfki-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-tr_TR-fahrettin-medium": _get_vits_piper,
-}
-
-vietnamese_models = {
-    "csukuangfj/vits-piper-vi_VN-25hours_single-low": _get_vits_piper,
-    "csukuangfj/vits-piper-vi_VN-vais1000-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-vi_VN-vivos-x_low": _get_vits_piper,
-    "csukuangfj/vits-mimic3-vi_VN-vais1000_low": _get_vits_piper,
-}
-
-bulgarian_models = {
-    "csukuangfj/vits-coqui-bg-cv": _get_vits_piper,
-}
-
-estonian_models = {
-    "csukuangfj/vits-coqui-et-cv": _get_vits_piper,
-}
-
-irish_models = {
-    "csukuangfj/vits-coqui-ga-cv": _get_vits_piper,
-}
-
-croatian_models = {
-    "csukuangfj/vits-coqui-hr-cv": _get_vits_piper,
-}
-
-lithuanian_models = {
-    "csukuangfj/vits-coqui-lt-cv": _get_vits_piper,
-}
-
-latvian_models = {
-    "csukuangfj/vits-coqui-lv-cv": _get_vits_piper,
-}
-
-maltese_models = {
-    "csukuangfj/vits-coqui-mt-cv": _get_vits_piper,
-}
-
-slovenian_models = {
-    "csukuangfj/vits-piper-sl_SI-artur-medium": _get_vits_piper,
-    "csukuangfj/vits-coqui-sl-cv": _get_vits_piper,
-}
-
-# Bangla
-bengali_models = {
-    "csukuangfj/vits-coqui-bn-custom_female": _get_vits_piper,
-    "csukuangfj/vits-mimic3-bn-multi_low": _get_vits_piper,
-}
-
-min_nan_models = {
-    "csukuangfj/vits-mms-nan": _get_vits_mms,
-}
-
-thai_models = {
-    "csukuangfj/vits-mms-tha": _get_vits_mms,
-}
-
-persian_models = {
-    "csukuangfj/vits-piper-fa_IR-amir-medium": _get_vits_piper,
-    "csukuangfj/vits-piper-fa_IR-gyro-medium": _get_vits_piper,
-    "csukuangfj/vits-mimic3-fa-haaniye_low": _get_vits_piper,
-}
-
-korean_models = {
-    "csukuangfj/vits-mimic3-ko_KO-kss_low": _get_vits_piper,
-}
-
-
-afrikaans_models = {
-    "csukuangfj/vits-mimic3-af_ZA-google-nwu_low": _get_vits_piper,
-}
-
-gujarati_models = {
-    "csukuangfj/vits-mimic3-gu_IN-cmu-indic_low": _get_vits_piper,
-}
-
-tswana_models = {
-    "csukuangfj/vits-mimic3-tn_ZA-google-nwu_low": _get_vits_piper,
-}
-
-
 language_to_models = {
     "Spanish": list(spanish_models.keys()),
 }