RobertaPipeline.py

import numpy as np
from transformers import QuestionAnsweringPipeline

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