diff --git a/models/ner.py b/models/ner.py index 36f9c3f..70d7d2f 100644 --- a/models/ner.py +++ b/models/ner.py @@ -23,6 +23,7 @@ ) from transformers import AdamW, get_linear_schedule_with_warmup + from label_studio_ml.model import LabelStudioMLBase from utils import calc_slope