Update app.py

app.py

@@ -68,15 +68,22 @@ class FluxGenerator:
 #            )
 
             
+#ERREUR `add_prefix_space`. The tokenizer needs to be converted from the slow tokenizers
+#            self.pipeline = FluxPipeline.from_pretrained(
+#                "black-forest-labs/FLUX.1-schnell",
+#                revision="refs/pr/1",
+#                device_map="balanced",  # Utilise une chaîne de caractères au lieu d'un dictionnaire
+#                **model_kwargs
+#            )        
+            
+            
             self.pipeline = FluxPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-schnell",
                 revision="refs/pr/1",
-                device_map="balanced",  # Utilise une chaîne de caractères au lieu d'un dictionnaire
-                **model_kwargs
-            )        
-            
-            
-            
+                device_map="auto",
+                torch_dtype=torch.float32,
+                use_safetensors=True
+)            
             
             
             
@@ -170,4 +177,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()