app.py

import gradio as gr
from transformers import AutoModelForCausalLM, AutoTokenizer

# Cargar el modelo y el tokenizer desde Hugging Face
model = AutoModelForCausalLM.from_pretrained("JuanCabs/lapepav0")
tokenizer = AutoTokenizer.from_pretrained("JuanCabs/lapepav0")

# Definir la función para generar texto
def generate_text(prompt):
    inputs = tokenizer(prompt, return_tensors="pt")  # Tokenizar la entrada
    outputs = model.generate(**inputs, max_length=100)  # Generar la salida
    return tokenizer.decode(outputs[0], skip_special_tokens=True)  # Decodificar y devolver el texto

# Crear la interfaz de Gradio
interface = gr.Interface(
    fn=generate_text,  # Función a ejecutar
    inputs="text",     # Tipo de entrada
    outputs="text",    # Tipo de salida
    title="Text Generation Model",  # Título de la app
    description="Enter a prompt to generate text using a Hugging Face model"  # Descripción de la app
)

# Lanzar la aplicació
interface.launch()