File size: 563 Bytes
6363086
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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()