From 5f8fd8b7a2f85305d126ad6385ac108c9c8946e0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bart=C5=82omiej=20Bojanowski?=
<bartlomiej.bojanowski@pwr.edu.pl>
Date: Mon, 21 Sep 2020 12:25:54 +0200
Subject: [PATCH] Fix pep8 format
---
src/predict_lang.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/src/predict_lang.py b/src/predict_lang.py
index 8f5b0bb..ef03443 100644
--- a/src/predict_lang.py
+++ b/src/predict_lang.py
@@ -1,15 +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',' ')
- print(text)
+ text = f.read().replace('\n', ' ')
predict = self.model.predict(text)[0][0]
with open(output_file, "w") as f:
f.write(predict)
--
GitLab