qa_roberta / app.py
antjohns's picture
Update app.py
e93e5fd verified
Raw History Blame Contribute Delete
430 Bytes
from transformers import pipeline
import gradio as gr
mdl_name = "deepset/roberta-base-squad2"
my_pipeline = pipeline('question-answering', model=mdl_name, tokenizer=mdl_name)
def answer_question(question, context):
inputs = {'question': question, 'context': context}
response = my_pipeline(inputs)
return response['answer']
gr.Interface(answer_question, inputs=["text", "text"], outputs="text").launch(share=True)