Update app.py

app.py

@@ -108,13 +108,20 @@ COMPOSITION_PARAMS = {
 }
 
 class ImageGenerator:
-     def __init__(self):
-        self.API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
-        token = os.getenv('HUGGINGFACE_TOKEN')
-        if not token:
-            logger.error("HUGGINGFACE_TOKEN non trouvé!")
-        self.headers = {"Authorization": f"Bearer {token}"}
-        logger.info("ImageGenerator initialisé")
+    def __init__(self):
+        self.API_URL = (
+            f"https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+            )
+        
+       # Chargement du token Hugging Face depuis les variables d'environnement
+       token = os.getenv('HUGGINGFACE_TOKEN')
+       
+       if not token:
+           logger.error("HUGGINGFACE_TOKEN non trouvé!")
+       
+       self.headers = {"Authorization": f"Bearer {token}"}
+       
+       logger.info("ImageGenerator initialisé")
 
     def _build_prompt(self, params: Dict[str, Any]) -> str:
        """Construction de prompt améliorée"""