File size: 627 Bytes
868a964 644ccb4 868a964 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import gradio as gr
from transformers import pipeline
pipe = pipeline("question-answering", model="huggingface-course/bert-finetuned-squad")
def qa(question, context):
return pipe(question=question, context=context)
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')
submit_btn.click(qa, inputs = [question, context], outputs = answer)
demo.launch()
|