app.py

# Importar Gradio
import gradio as gr
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="microsoft/phi-1_5", trust_remote_code=True)

# Función para generar texto
def generar_texto(prompt):
    # Usar el pipeline para generar texto
    generated_text = pipe(prompt, max_length=50, do_sample=True)
    return generated_text[0]['generated_text']

# Crear la interfaz de Gradio
iface = gr.Interface(
    fn=generar_texto,
    inputs=gr.inputs.Textbox(default="Ingresa tu prompt aquí..."),
    outputs=gr.outputs.Textbox(label="Texto generado"),
    live=True,
    title="Generador de Texto",
    description="Ingresa un prompt y obtén texto generado."
)

# Ejecutar la interfaz
iface.launch()