Spaces:
Runtime error
Runtime error
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() |