This commit is contained in:
Patrick Jentsch
2019-05-20 12:08:13 +02:00
parent 5b7bc2a840
commit 6c8b32fad4
4 changed files with 67 additions and 29 deletions

View File

@ -15,7 +15,7 @@ parser.add_argument(
)
parser.add_argument(
'-l',
choices=['de', 'en', 'es', 'fr', 'pt'],
choices=['de', 'el', 'en', 'es', 'fr', 'it', 'nl', 'pt'],
dest='lang',
required=True
)
@ -26,8 +26,9 @@ parser.add_argument(
args = parser.parse_args()
SPACY_MODELS = {
'de': 'de_core_news_sm', 'en': 'en_core_web_sm', 'es': 'es_core_news_sm',
'fr': 'fr_core_news_sm', 'pt': 'pt_core_news_sm'
'de': 'de_core_news_sm', 'el': 'el_core_news_sm', 'en': 'en_core_web_sm',
'es': 'es_core_news_sm', 'fr': 'fr_core_news_sm', 'it': 'it_core_news_sm',
'nl': 'nl_core_news_sm', 'pt': 'pt_core_news_sm'
}
# Set the language model for spacy