diff --git a/src/speller2_worker.py b/src/speller2_worker.py index 95efe002ec29eb0e154acc91442aab55b22d10e3..39f1b23ff9bc895264b0ec21f3635663cf9cc3c6 100755 --- a/src/speller2_worker.py +++ b/src/speller2_worker.py @@ -14,14 +14,17 @@ class Speller2Worker(nlp_ws.NLPWorker): def static_init(cls, config): """One time static initialisation.""" _log.log(logging.INFO, "Worker started loading static models ") - cls._model = Speller('pl') + cls._model = {'pl': Speller('pl'), + 'ru': Speller('ru'), + 'en': Speller('en'), + 'uk': Speller('uk')} _log.log(logging.INFO, "Worker finished loading static models ") def process(self, input_file, task_options, output_file): """Starting nlp process.""" _log.info("Processing") - - model = self._model + language = task_options.get('lang', 'pl') + model = self._model.get(language) data = self._read_file(input_file)