import gradio as gr
from transformers import pipeline

pipe = pipeline("question-answering", model="deepset/roberta-base-squad2-distilled", framework="pt")

def chat(message, history):
    context="A laranja é uma fruta laranja e o limão é uma fruta verde"
    resposta = pipe(question=message, context=context)
    return resposta['answer']


demo = gr.ChatInterface(chat)

if __name__ == "__main__":
    demo.launch()