solucion de posible errores

app.py

@@ -44,8 +44,7 @@ class ModelHandler:
 
 # Lista de modelos disponibles
 model_names = [
-    "microsoft/Phi-3-mini-4k-instruct",
-    "bigscience/bloomz-560m"
+    "microsoft/Phi-3-mini-4k-instruct"
 ]
 
 # Inicializa el manejador de modelos
@@ -57,10 +56,9 @@ def generate_image(prompt):
     Genera una imagen utilizando un modelo de generación de imágenes.
     """
     try:
-        # Aquí puedes usar el modelo de Hugging Face para generación de imágenes.
-        # Ejemplo ficticio:
-        image_url = f"https://via.placeholder.com/512.png?text={prompt.replace(' ', '+')}"
-        return image_url
+        client = InferenceClient("CompVis/stable-diffusion-v1-4", token=hf_token)
+        image = client.text_to_image(prompt, width=512, height=512)
+        return image
     except Exception as e:
         return f"Error al generar la imagen: {e}"
 
@@ -118,4 +116,4 @@ with gr.Blocks(title="Multi-Model LLM Chatbot with Image Generation") as demo:
     )
 
 # Lanza la interfaz
-demo.launch()
+demo.launch()