Liabot / app.py
Vitors's picture
Update app.py
d330844 verified
raw
history blame contribute delete
686 Bytes
import gradio as gr
from transformers import pipeline, set_seed
set_seed(42)
# Modelo em português 100% funcional
modelo = pipeline("text-generation", model="pierreguillou/gpt2-small-portuguese")
# Função do chatbot
def responder(pergunta):
resposta = modelo(pergunta, max_length=80, num_return_sequences=1)[0]['generated_text']
return resposta
# Interface Gradio
gr.Interface(
fn=responder,
inputs=gr.Textbox(label="Digite sua pergunta em português"),
outputs=gr.Textbox(label="Resposta do modelo"),
title="Chatbot GPT-2 em Português 🇧🇷",
description="Este chatbot responde perguntas usando o modelo GPT-2 treinado em português."
).launch()