Spaces:
Runtime error
Runtime error
import gradio as gr | |
import spaces | |
from transformers import pipeline | |
pipe = pipeline(task="question-answering", model="distilbert/distilbert-base-cased-distilled-squad") | |
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() |