import gradio as gr
from transformers import pipeline

# Carrega o modelo (modifique para o modelo que você usa)
modelo = pipeline("text-generation", model="gpt2")

# Função para processar a entrada do usuário
def gerar_texto(entrada):
    resultado = modelo(entrada, max_length=50)
    return resultado[0]["generated_text"]

# Interface do Gradio
iface = gr.Interface(
    fn=gerar_texto,
    inputs=gr.Textbox(label="Digite seu texto"),
    outputs=gr.Textbox(label="Texto Gerado"),
)

# Lançar app (sem `share=True`)
iface.launch()