Commit f7aef48c authored by Vít Novotný's avatar Vít Novotný
Browse files

Fix type check

parent 7770581d
......@@ -45,6 +45,7 @@ class NerModel:
@property
def model(self) -> AutoModelForTokenClassification:
model = AutoModelForTokenClassification.from_pretrained(self.model_name_or_basename)
return model
def __str__(self) -> str:
return self.model_name_or_basename
......
Supports Markdown
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