pipeline.py

from transformers import QuestionAnsweringPipeline

class ExtractiveQAPipeline(QuestionAnsweringPipeline):
    def postprocess(self, model_outputs):
        answers = super().postprocess(model_outputs)
        
        return {'guess': answers['answer'], 'confidence': answers['score']}