|
import gradio as gr |
|
from transformers import pipeline, set_seed |
|
|
|
set_seed(42) |
|
|
|
|
|
modelo = pipeline("text-generation", model="pierreguillou/gpt2-small-portuguese") |
|
|
|
|
|
def responder(pergunta): |
|
resposta = modelo(pergunta, max_length=80, num_return_sequences=1)[0]['generated_text'] |
|
return resposta |
|
|
|
|
|
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() |
|
|