first-draft qa

1draft.py

from allennlp.predictors import Predictor
from transformers.tokenization_gpt2 import GPT2Tokenizer
from transformers import pipeline

class HuggingFacePredictor(Predictor):
    def __init__(self) -> None:
        self.tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
        self.model = pipeline('question-answering')

    def predict(self, passage='', question=''):
        return self.model({
            'context': passage,
            'question': question
        })

    def __getitem__(self, index: int) -> str:
        return self.tokenizer.decode([index])