File size: 686 Bytes
7b5143c b3d3e88 937f457 b3d3e88 937f457 d330844 b3d3e88 d330844 89bde4d d330844 937f457 b3d3e88 89bde4d d330844 b3d3e88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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()
|