Commit 4dfbf7e1 authored by Jarema Radom's avatar Jarema Radom

style fix

parent 645b7882
Pipeline #3192 passed with stages
in 4 minutes and 52 seconds
......@@ -109,8 +109,8 @@ class Worker(nlp_ws.NLPWorker):
f.write(text_out)
def initialize_model(self, model_path: str):
self.tool = language_tool_python.LanguageTool(
self.languagetool_map[model_path])
self.tool = language_tool_python.LanguageTool(self.languagetool_map
[model_path])
if self.model:
self.model.to('cpu')
self.model = AutoModelForTokenClassification.from_pretrained(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment