diff --git a/src/speller2_worker.py b/src/speller2_worker.py index 6bc142e5e580b2e97205187b84e6a5daa07f78f4..172972302864bbcc8a99361b512b6a2e248084aa 100755 --- a/src/speller2_worker.py +++ b/src/speller2_worker.py @@ -26,11 +26,15 @@ class Speller2Worker(nlp_ws.NLPWorker): model = self._model data = self._read_file(input_file) + print(data.split('\n')) + corrected_data = [model(line) for line in data.split('\n')] - corrected_data = model(data) + with open(output_file, 'w', encoding='utf-8') as f: + f.write('\n'.join(corrected_data)) - with io.open(output_file, 'w', encoding='utf-8') as f: - json.dump(corrected_data, f, ensure_ascii=False) + #with io.open(output_file, 'w', encoding='utf-8') as f: + #json.dump(corrected_data, f, ensure_ascii=False) + #for line in corrected_data: @classmethod def _read_file(cls, input_path):