Update app.py

app.py

@@ -57,12 +57,30 @@ class FluxGenerator:
                 "use_safetensors": True,  # Utilise safetensors pour un chargement plus efficace
             }
 
+            
+#Erreur   Erreur lors du chargement du modèle: `device_map` must be a string.          
+#
+#            self.pipeline = FluxPipeline.from_pretrained(
+#                "black-forest-labs/FLUX.1-schnell",
+#                revision="refs/pr/1",
+#                device_map={"": self.device},
+#                **model_kwargs
+#            )
+
+            
             self.pipeline = FluxPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
-                device_map={"": self.device},
+                device_map="cpu",  # Utilise une chaîne de caractères au lieu d'un dictionnaire
                 **model_kwargs
-            )
+            )        
+            
+            
+            
+            
+            
+            
+            
 
             # Optimisations supplémentaires
             self.pipeline.enable_sequential_cpu_offload()  # Décharge les composants non utilisés
@@ -152,4 +170,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()