File size: 412 Bytes
edd3254
 
 
 
 
2ca9b3b
edd3254
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import gradio as gr
import spaces

from transformers import pipeline

pipe = pipeline(task="question-answering", model="distilbert/distilbert-base-cased-distilled-squad")

@spaces.GPU
def get_answer(context, question):
  result = pipe(context=context, question=question)
  return result['answer']

demo = gr.Interface(fn=get_answer, inputs=["textarea", "text"], outputs=gr.Textbox(label="answer"))

demo.launch()