From 15cca73b49cc9eb46ee1416f4d0af2a4f7765ada Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bart=C5=82omiej=20Bojanowski?=
<bartlomiej.piotr.bojanowski@gmail.com>
Date: Tue, 25 May 2021 10:02:10 +0200
Subject: [PATCH] Add en, ru, uk language.
---
src/speller2_worker.py | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/src/speller2_worker.py b/src/speller2_worker.py
index 95efe00..39f1b23 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)
--
GitLab