| import gradio as gr | |
| import torch | |
| from transformers import pipeline | |
| modelo = pipeline("text-to-image", model="runwayml/stable-diffusion-v1-5") | |
| def generarImagen(texto): | |
| imagen = modelo(prompt=texto).images[0] | |
| return imagen | |
| with gr.Blocks() as demo: | |
| gr.Markdown("# Generador Visual de Historia") | |
| gr.Markdown("Introduce la historia a la que quieras generar el texto") | |
| with gr.Row(): | |
| with gr.Column(scale=1): | |
| entrada = gr.Textbox( | |
| label="Fragmento de la Historia", | |
| placeholder="Introduce" | |
| ) | |
| btn = gr.Button("Generar Imagen", variant="primary") | |
| with gr.Column(scale=1): | |
| salida = gr.Image(label="Imagen Generada", type="pil") | |
| btn.click(fn=generarImagen, inputs=entrada, outputs=salida) | |
| demo.launch(theme=gr.themes.Monochrome()) | |