File size: 759 Bytes
ea72a26 c3467af 88a0841 c3467af ea72a26 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# 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()
|