Update interfaces/base_pipeline.py

interfaces/base_pipeline.py

@@ -29,7 +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  = TokenClassificationPipeline(device=device=torch.device("cpu"), model=model, tokenizer=tokenizer, aggregation_strategy = 'first')
+transformers_pipeline  = TokenClassificationPipeline(model=model, tokenizer=tokenizer, aggregation_strategy = 'first')
 
 if __name__=="__main__":
     pipeline = generate_pipeline()