| 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']} |