fix(inferer): fix missing var

microservices/inferer/inferer.py

@@ -125,9 +125,9 @@ def inference_process(inference_data, model_id):
     model = RobertaForSpanCategorization.from_pretrained(model_id)
     tokenizer = AutoTokenizer.from_pretrained(model_id)
 
-    def get_offsets_and_predicted_tags(example: str, model, tokenizer, threshold=0):
-        raw_encoded_example = tokenizer(example, return_offsets_mapping=True)
-        encoded_example = tokenizer(example, return_tensors="pt")
+    def get_offsets_and_predicted_tags(sentence: str, model, tokenizer, threshold=0):
+        raw_encoded_example = tokenizer(sentence, return_offsets_mapping=True)
+        encoded_example = tokenizer(sentence, return_tensors="pt")
         out = model(**encoded_example)["logits"][0]
         predicted_tags = [[i for i, l in enumerate(logit) if l > threshold] for logit in out]