diff --git a/Dockerfile b/Dockerfile index 4216c34d84b58089d1b521309301ed51b18b2e13..9f354ba24fbd671c4687800d7bcb4580c5ad8b73 100644 --- a/Dockerfile +++ b/Dockerfile @@ -25,7 +25,7 @@ RUN poetry run python -m spacy download en_core_web_lg RUN poetry run python -m spacy download it_core_news_lg RUN poetry run python -m spacy download nl_core_news_lg RUN poetry run python -m spacy download sp_core_news_lg -RUN poetry run python -m spacy download pt_core_news_lg +RUN poetry run python -m spacy download es_core_news_lg poetry run python -m spacy download de_core_news_lg poetry run python -m spacy download pl_core_news_lg @@ -33,4 +33,4 @@ poetry run python -m spacy download en_core_web_lg poetry run python -m spacy download it_core_news_lg poetry run python -m spacy download nl_core_news_lg poetry run python -m spacy download sp_core_news_lg -poetry run python -m spacy download pt_core_news_lg +poetry run python -m spacy download es_core_news_lg diff --git a/new_experiment/utils/get_spacy_model_name.py b/new_experiment/utils/get_spacy_model_name.py index bf910a63c6df634f1634ff08a719532d444a1d05..a43b504613500274ed4a9863361ba7351323c8f2 100644 --- a/new_experiment/utils/get_spacy_model_name.py +++ b/new_experiment/utils/get_spacy_model_name.py @@ -1,2 +1,4 @@ def get_spacy_model_name(language_code_2_letter: str) -> str: + if language_code_2_letter == 'en': + return 'en_core_web_lg' return f'{language_code_2_letter}_core_news_lg'