custom_pipeline.py

from transformers import QuestionAnsweringPipeline, pipeline, AutoModelForQuestionAnswering, TFAutoModelForQuestionAnswering, AutoTokenizer

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