from transformers import TextClassificationPipeline class PipelinePreparer: @staticmethod def prepare_pipeline(tokenizer, model): """Create text classification pipeline""" return TextClassificationPipeline( tokenizer=tokenizer, model=model, return_all_scores=True )