From 773465c31f6fd366e4b8749301b27d7ba891dbd4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bart=C5=82omiej=20Bojanowski?= Date: Fri, 25 Sep 2020 12:09:06 +0200 Subject: [PATCH] Modify README and fix PredictLangWorker --- README.md | 2 +- src/predict_lang.py | 15 +++++++-------- 2 files changed, 8 insertions(+), 9 deletions(-) diff --git a/README.md b/README.md index fedacb2..6ad49ce 100644 --- a/README.md +++ b/README.md @@ -9,6 +9,6 @@ Use docker to build and run image: ``` docker build -t . -docker run -v "/samba:/samba" +docker run -v /samba:/samba ``` diff --git a/src/predict_lang.py b/src/predict_lang.py index ef03443..99c319a 100644 --- a/src/predict_lang.py +++ b/src/predict_lang.py @@ -1,18 +1,17 @@ import fasttext import nlp_ws - class PredictLangWorker(nlp_ws.NLPWorker): - + def __init__(self): self.model = fasttext.load_model("lid.176.bin") - def process(self, input_file: str, - task_options: dict, - output_file: str) -> None: + def process(self, input_file: str, task_options: dict, output_file: str) -> None: with open(input_file, "r") as f: - text = f.read().replace('\n', ' ') + text = f.read().replace('\n',' ') + print(text) predict = self.model.predict(text)[0][0] with open(output_file, "w") as f: - f.write(predict) - return self.model.predict(str(text)) + f.write(text+ predict) + print(predict) + return predict -- GitLab