Update app.py

app.py

@@ -71,7 +71,7 @@ class FluxGenerator:
             self.pipeline = FluxPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
-                device_map="cpu",  # Utilise une chaîne de caractères au lieu d'un dictionnaire
+                device_map="balanced",  # Utilise une chaîne de caractères au lieu d'un dictionnaire
                 **model_kwargs
             )