components/pipeline_preparer.py

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
        )