Iosif24/qa_roberta

1from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline2import gradio as grad3import ast4 5model_name = "deepset/roberta-base-squad2"6my_pipeline = pipeline('question-answering', model=model_name, tokenizer=model_name)7 8def answer_question(question, context):9    text = {'question':question, 'context': context}10    response = my_pipeline(text)11    return response12grad.Interface(answer_question, inputs=["text","text"], outputs="text").launch()