Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
# Carregando o modelo de chatbot do Hugging Face | |
model = pipeline("text-chat", device=0) | |
def chatbot_conversation(input_text): | |
# Resposta do chatbot | |
response = model(input_text)[0]['generated_text'] | |
return response | |
# Interface do Gradio | |
iface = gr.Interface( | |
fn=chatbot_conversation, | |
inputs=gr.Textbox(lines=2, label="Pergunta"), | |
outputs=gr.Textbox(label="Resposta"), | |
title="Chatbot Simples", | |
description="Converse com o chatbot!" | |
) | |
# Executando a interface | |
iface.launch() | |