diff --git a/docker-compose.yml b/docker-compose.yml index 0b1e0e60190891e661cb070dcd9ae414966135ae..933e03cb00e35c8fc98be68dfb18cf64f48c9a19 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -15,3 +15,4 @@ services: - './config.ini:/home/worker/config.ini' - './src:/home/worker/src' - './main.py:/home/worker/main.py' + - './info.json:/home/worker/info.json' diff --git a/src/info.json b/info.json similarity index 100% rename from src/info.json rename to info.json diff --git a/src/omwn_worker.py b/src/omwn_worker.py index dbd8542808f405541040cefa7c11cf41b12d1785..1c01a9bc04462e04616e4af8db7893a12caa44dd 100644 --- a/src/omwn_worker.py +++ b/src/omwn_worker.py @@ -1,8 +1,10 @@ -"""Implementation of omwn_worker.""" +"""Implementation of omwn_worker.""" import logging import json + import lex_ws +import nltk from nltk.corpus import wordnet as wn my_logger = logging.getLogger(__name__) @@ -17,6 +19,8 @@ class OMWNWorker(lex_ws.LexWorker): @classmethod def static_init(cls, config): """Initialize worker.""" + nltk.download('wordnet') + nltk.download('omw') my_logger.info("Loading models...") for lang in languages: wn.lemmas("test", lang=languages[lang])