chatmgf / app.py
diogocarapito's picture
Update app.py
63d1441
raw
history blame
478 Bytes
import gradio as gr
from transformers import pipeline
model_name = "deepset/roberta-base-squad2"
pipe = pipeline("question-answering", model=model_name, tokenizer=model_name)
context = "my name is diogo and i'm 32 years old, i live in portugal neraby the ocean."
def q_and_a(text):
query = {
'question': text,
'context': context
}
return pipe(query)['answer']
interface = gr.Interface(
fn=q_and_a,
inputs='text',
outputs='text'
)
interface.launch()