Skip to content
Snippets Groups Projects
Commit 2e93a16a authored by Michal Pogoda's avatar Michal Pogoda
Browse files

Fixed config variables beeing string instead of numeric

parent c6bf26dc
No related branches found
No related tags found
1 merge request!3Hotfixes
Pipeline #1306 passed
......@@ -15,8 +15,8 @@ class Worker(nlp_ws.NLPWorker):
self.config = configparser.ConfigParser()
self.config.read("config.ini")
self.threshold = self.config["deployment"]["threshold"]
self.chunk_size = self.config["deployment"]["chunk_size"]
self.threshold = float(self.config["deployment"]["threshold"])
self.chunk_size = int(self.config["deployment"]["chunk_size"])
self.tokenizer, self.model = load_model(
self.config["deployment"]["model"],
self.config["deployment"]["base_model"],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment