Create app.py

app.py

@@ -0,0 +1,57 @@
+import gradio as gr
+from diffusers import DiffusionPipeline
+import torch
+
+# Cargar el modelo
+model_id = "sd-community/sdxl-flash"
+pipeline = DiffusionPipeline.from_pretrained(model_id)
+
+# Estilos CSS personalizados
+custom_css = """
+<style>
+body {
+    background-color: #1f1f1f;
+    color: #ffffff;
+    font-family: Arial, sans-serif;
+}
+.gr-form input[type="text"], .gr-form select {
+    background-color: #333333 !important;
+    border: 1px solid #777777 !important;
+    color: #ffffff !important;
+}
+.gr-output {
+    background-color: #333333 !important;
+    border: 2px solid #777777 !important;
+    border-radius: 5px;
+    padding: 10px;
+    margin-top: 20px;
+}
+.gr-control.gr-number input {
+    width: 100px !important;
+}
+</style>
+"""
+
+def generate_image(prompt, steps, negative_prompt, num_images):
+    result = pipeline(prompt, num_inference_steps=steps, negative_prompt=negative_prompt, num_images=num_images)
+    images = result.images
+    return images
+
+# Crear la interfaz de Gradio
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=[
+        gr.inputs.Textbox(label="Prompt", placeholder="Ingrese un prompt...", lines=5),
+        gr.inputs.Number(label="Steps", default=10, min=1, max=15),
+        gr.inputs.Textbox(label="Negative Prompt (Opcional)", placeholder="Ingrese un negative prompt...", lines=2, optional=True),
+        gr.inputs.Number(label="Número de Imágenes", default=1, min=1, max=8)
+    ],
+    outputs=gr.outputs.Image(label="Imágenes Generadas"),
+    title="Generador de Imágenes con Stable Diffusion",
+    description="Este modelo de Stable Diffusion genera imágenes a partir de un prompt.",
+    theme="gradio",
+    css=custom_css
+)
+
+# Ejecutar la interfaz
+iface.launch()