Update interfaces/base_pipeline.py

interfaces/base_pipeline.py

@@ -29,8 +29,7 @@ def generate_pipeline(threshold: float = 0.5):
 
 tokenizer = AutoTokenizer.from_pretrained("knowledgator/UTC-DeBERTa-large-v2")
 model = AutoModelForTokenClassification.from_pretrained("knowledgator/UTC-DeBERTa-large-v2")
-
-transformers_pipeline  = pipeline(device=torch.device("cpu"), task="ner", model ="knowledgator/UTC-DeBERTa-small", pipeline_class =TokenClassificationPipeline, aggregation_strategy = "first", batch_size = 12)
+transformers_pipeline  = TokenClassificationPipeline(device=device=torch.device("cpu"), model=model, tokenizer=tokenizer, aggregation_strategy = 'first')
 
 if __name__=="__main__":
     pipeline = generate_pipeline()