Skip to content
Snippets Groups Projects
get_spacy_model_name.py 262 B
Newer Older
Marcin Wątroba's avatar
Marcin Wątroba committed
import spacy


Marcin Wątroba's avatar
Marcin Wątroba committed
def get_spacy_model_name(language_code_2_letter: str) -> str:
Marcin Wątroba's avatar
Marcin Wątroba committed
    if language_code_2_letter == 'en':
        return 'en_core_web_lg'
Marcin Wątroba's avatar
Marcin Wątroba committed
    return f'{language_code_2_letter}_core_news_lg'
Marcin Wątroba's avatar
Marcin Wątroba committed


if __name__ == '__main__':
    spacy.load('en_core_web_lg')