From 773465c31f6fd366e4b8749301b27d7ba891dbd4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bart=C5=82omiej=20Bojanowski?=
<bartlomiej.bojanowski@pwr.edu.pl>
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 <name> .
-docker run <name> -v "/samba:/samba"
+docker run -v /samba:/samba <name>
```
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