def __init__(self, lang: str = 'en', cache_dir: str = None, overwrite_cache: bool = False): if lang == 'en': from nemo_text_processing.inverse_text_normalization.en.taggers.tokenize_and_classify import ClassifyFst from nemo_text_processing.inverse_text_normalization.en.verbalizers.verbalize_final import ( VerbalizeFinalFst, ) elif lang == 'es': from nemo_text_processing.inverse_text_normalization.es.taggers.tokenize_and_classify import ClassifyFst from nemo_text_processing.inverse_text_normalization.es.verbalizers.verbalize_final import ( VerbalizeFinalFst, ) elif lang == 'ru': from nemo_text_processing.inverse_text_normalization.ru.taggers.tokenize_and_classify import ClassifyFst from nemo_text_processing.inverse_text_normalization.ru.verbalizers.verbalize_final import ( VerbalizeFinalFst, ) elif lang == 'de': from nemo_text_processing.inverse_text_normalization.de.taggers.tokenize_and_classify import ClassifyFst from nemo_text_processing.inverse_text_normalization.de.verbalizers.verbalize_final import ( VerbalizeFinalFst, ) elif lang == 'fr': from nemo_text_processing.inverse_text_normalization.fr.taggers.tokenize_and_classify import ClassifyFst from nemo_text_processing.inverse_text_normalization.fr.verbalizers.verbalize_final import ( VerbalizeFinalFst, ) self.tagger = ClassifyFst(cache_dir=cache_dir, overwrite_cache=overwrite_cache) self.verbalizer = VerbalizeFinalFst() self.parser = TokenParser()
def __init__(self, lang: str = 'en'): if lang == 'en': from nemo_text_processing.inverse_text_normalization.en.taggers.tokenize_and_classify import ClassifyFst from nemo_text_processing.inverse_text_normalization.en.verbalizers.verbalize_final import ( VerbalizeFinalFst, ) self.tagger = ClassifyFst() self.verbalizer = VerbalizeFinalFst() self.parser = TokenParser()