Spaces:
Running
Running
from bert_vits2.text import chinese, cleaned_text_to_sequence | |
language_module_map = { | |
'ZH': chinese | |
} | |
def clean_text(text, language): | |
language_module = language_module_map[language] | |
norm_text = language_module.text_normalize(text) | |
phones, tones, word2ph = language_module.g2p(norm_text) | |
return norm_text, phones, tones, word2ph | |
def clean_text_bert(text, language): | |
language_module = language_module_map[language] | |
norm_text = language_module.text_normalize(text) | |
phones, tones, word2ph = language_module.g2p(norm_text) | |
bert = language_module.get_bert_feature(norm_text, word2ph) | |
return phones, tones, bert | |
def text_to_sequence(text, language): | |
norm_text, phones, tones, word2ph = clean_text(text, language) | |
return cleaned_text_to_sequence(phones, tones, language) | |
if __name__ == '__main__': | |
pass | |