| import gradio as gr | |
| from transformers import pipeline | |
| pipe = pipeline("question-answering", model="huggingface-course/bert-finetuned-squad") | |
| def qa(question, context): | |
| result = pipe(question=question, context=context) | |
| return result['answer'], result['score'] | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| with gr.Column(): | |
| question = gr.Textbox(label= 'Question') | |
| context = gr.Textbox(label= 'Context') | |
| submit_btn = gr.Button(value = 'Submit') | |
| with gr.Column(): | |
| answer = gr.Textbox(label= 'Answer') | |
| score = gr.Textbox(label= 'Score') | |
| submit_btn.click(qa, inputs = [question, context], outputs = [answer, score]) | |
| demo.launch() | |