otro

app.py

@@ -1,6 +1,10 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
 import torch
+from transformers.utils import move_cache
+
+# Mover la caché si es necesario
+move_cache()
 
 # Cargar el modelo base y los pesos LoRA
 model_name = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -18,17 +22,27 @@ def generate_image(prompt):
         image = pipeline(prompt).images[0]
     return image
 
-# Crear la interfaz de Gradio
+# Crear la interfaz de Gradio con CSS personalizado
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.inputs.Textbox(lines=5, label="Descripción de la imagen", placeholder="Introduce el texto aquí..."),
-    outputs=gr.outputs.Image(type="pil", label="Imagen generada"),
+    inputs=gr.Textbox(lines=5, label="Descripción de la imagen", placeholder="Introduce el texto aquí..."),
+    outputs=gr.Image(type="pil", label="Imagen generada"),
     title="Generador de Imágenes a partir de Texto",
     description="Ingrese un texto y obtenga una imagen generada por IA.",
     examples=[
         ["A picturesque view of a local market. The first light of day illuminates the stone facades and worn tiles of the houses and buildings, some of which date back centuries. At the center of the scene, a cobblestone square leads to an open-air market that begins to come to life, with vendors setting up their stalls selling fruits, vegetables, flowers, and local crafts. The narrow, winding streets are lined with old lanterns, now unlit, while lazy cats lounge on the stone steps. In one corner, an ancient fountain, adorned with weathered carvings, murmurs softly, adding to the tranquil atmosphere. In the background, the towers of an ancient cathedral rise, capturing the first rays of sunlight that paint the sky in soft pinks and oranges. This image should convey a sense of tranquility, beauty, and a deep connection to the past, celebrating the rich history and timeless charm of the ancient village or town."]
     ],
-    live=True
+    live=True,
+    css="""
+    .gradio-container {
+        background-color: #f9f9f9;
+        color: #333;
+        font-family: Arial, sans-serif;
+    }
+    .gradio-container h1 {
+        color: #4CAF50;
+    }
+    """
 )
 
 # Ejecutar la interfaz