nuevo2

app.py

@@ -1,24 +1,22 @@
 import gradio as gr
-from transformers import AutoModelForCausalLM, AutoTokenizer
-from torch import autocast
+from diffusers import DiffusionPipeline
 import torch
 
-# Cargar el modelo y el tokenizer
-model_name = "ZB-Tech/Text-to-Image"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+# Cargar el modelo base y los pesos LoRA
+model_name = "stabilityai/stable-diffusion-xl-base-1.0"
+lora_weights = "ZB-Tech/Text-to-Image"
+pipeline = DiffusionPipeline.from_pretrained(model_name)
+pipeline.load_lora_weights(lora_weights)
 
+# Mover el modelo a la GPU si está disponible
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model = model.to(device)
+pipeline = pipeline.to(device)
 
 # Definir la función de generación de imagen
 def generate_image(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt").to(device)
-    with autocast(device):
-        output = model.generate(**inputs)
-    # Convertir el output en una imagen adecuada para mostrar
-    # (El código depende de la salida del modelo)
-    return tokenizer.decode(output[0], skip_special_tokens=True)
+    with torch.autocast(device):
+        image = pipeline(prompt).images[0]
+    return image
 
 # Crear la interfaz de Gradio
 iface = gr.Interface(
@@ -27,12 +25,10 @@ iface = gr.Interface(
     outputs=gr.outputs.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.",
-    theme="default",  # Puedes cambiar el tema
-    layout="vertical",  # Puedes cambiar el diseño a 'horizontal' o 'vertical'
     examples=[
-        ["a picturesque of 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."]
+        ["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  # Si deseas que los resultados se actualicen en tiempo real
+    live=True
 )
 
 # Ejecutar la interfaz