generate_image [FLUX_1.0 {Gracio Client Api (Huggingface spaces)}]

BorjaOteroFerreira · Sep 17, 2024 · a2e0f44 · a2e0f44
1 parent 49d3263
commit a2e0f44
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/tools/generate_image.py b/tools/generate_image.py
@@ -6,8 +6,7 @@
 from gradio_client import Client
 from huggingface_hub import login
 
-# Reemplaza 'your_token' por tu token de Hugging Face
-login(token="huggingface_token")
+
 
 
 class ImageGenerationResult(BaseModel):
@@ -50,7 +49,14 @@ def run(
         Returns:
             List[ImageGenerationResult]: Una lista con la URL del archivo generado y la semilla utilizada.
         """
+        try: 
+            login(token="huggingface_token", add_to_git_credential=True)
+        except Exception as e:
+            raise Exception("No se pudo iniciar sesión en Hugging Face Hub. Por favor, verifica tu token.", e)
+
         try:
+            # Reemplaza 'your_token' por tu token de Hugging Face
+
             # Crear el cliente de Gradio
             client = Client("black-forest-labs/FLUX.1-dev")
             # Realizar la predicción para generar la imagen
@@ -72,11 +78,11 @@ def run(
             destination_path = os.path.join(images_folder, file_name)
             # Mover la imagen desde la carpeta temporal a 'static/images'
             shutil.move(temp_file_path, destination_path)
-            # Devolver la URL de la imagen generada y la semilla
+            # Devolver la URL de la imagen generada
             imagen = f"<img src='static/tools/generate_image/{file_name}' width='590' height='345'>", 
-            respuesta = f'IMAGEN de {prompt} -> {imagen}, ES IMPORTANTISIMO QUE INCLUYAS LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta y sin acortar ni alterar la url'
+            respuesta = f'IMAGEN de {prompt} -> {imagen}, ES IMPORTANTISIMO QUE RESPONDAS SOLO CON LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta y sin acortar ni alterar la url'
             return str(respuesta)
         except Exception as e:
             print(e)
-            respuesta = f'IMAGEN de {prompt} : /static/tools/generate_image/pj.png, ES IMPORTANTISIMO QUE INCLUYAS LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta y sin acortar ni alterar la url'
+            respuesta = f"IMAGEN de {prompt} -> <img src='/static/tools/generate_image/pj.png' width='590' height='345'>, ES IMPORTANTISIMO QUE RESPONDAS SOLO CON LA ETIQUETA DE IMAGEN EN TU RESPUESTA RESPETANDO EL FORMATO,  sin texto adicional despues de la etiqueta. NO ALTERES LA URL"
             return respuesta