diff --git a/pipelines.py b/pipelines.py index 6813bf0..92733dc 100644 --- a/pipelines.py +++ b/pipelines.py @@ -1,4 +1,8 @@ -from transformers import AutoModelForTokenClassification, AutoTokenizer, TokenClassificationPipeline +from transformers import ( + AutoModelForTokenClassification, + AutoTokenizer, + TokenClassificationPipeline, +) from transformers.pipelines import AggregationStrategy import numpy as np