app.py

import gradio as gr
from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_pretrained(
    "stabilityai/sd-turbo",
    torch_dtype=torch.float32
)

pipe = pipe.to("cpu")

def generar_imagen(prompt):
    imagen = pipe(prompt, num_inference_steps=1).images[0]
    return imagen

with gr.Blocks() as demo:
    gr.Markdown("# 🎨 ImaginaAI")
    gr.Markdown("Genera imágenes con inteligencia artificial")

    prompt = gr.Textbox(label="Escribe tu idea")

    boton_generar = gr.Button("Generar imagen")
    resultado = gr.Image(label="Imagen generada")

    boton_generar.click(fn=generar_imagen, inputs=prompt, outputs=resultado)

demo.launch()