From dea0a8df30214ffe60514171ae84c34bd744881f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marcin=20W=C4=85troba?= <markowanga@gmail.com>
Date: Thu, 12 Jan 2023 08:10:44 +0100
Subject: [PATCH] Update spacy models
---
Dockerfile | 4 ++--
new_experiment/utils/get_spacy_model_name.py | 2 ++
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/Dockerfile b/Dockerfile
index 4216c34..9f354ba 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 bf910a6..a43b504 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'
--
GitLab