Update app.py

app.py

@@ -103,6 +103,30 @@ class PresentationGenerator:
 #        )
 
 ##Modif01 : Correction Pour Flux Non Chargé sur HFSpace
+#    def load_image_model(self, model_name):
+#        """Charge le modèle de génération d'images"""
+#        model_id = IMAGE_MODELS[model_name]
+#        if model_id == "black-forest-labs/FLUX.1-schnell":
+#            self.image_pipeline = FluxPipeline.from_pretrained(
+#                model_id,
+#                torch_dtype=torch.bfloat16
+#            )
+#            self.image_pipeline.enable_model_cpu_offload()  # Économise de la VRAM en déchargeant le modèle sur le CPU
+#            print(f"Modèle d'image FLUX chargé : {model_id}")
+#        else:
+#            self.image_pipeline = pipeline(
+#                "text-to-image",
+#                model=model_id,
+#                token=self.token
+#            )
+#        print(f"Modèle d'image chargé : {model_id}")
+
+
+##Modif02 : Correction Pour Flux Blocage Chargement a 71% sur HFSpace
+#Loading pipeline components...:  71%|███████▏  | 5/7 [00:05<00:01,  1.07it/s]You set `add_prefix_space`. 
+#The tokenizer needs to be converted from the slow tokenizers
+#Loading pipeline components...:  71%|███████▏  | 5/7 [00:05<00:02,  1.15s/it]
+
     def load_image_model(self, model_name):
         """Charge le modèle de génération d'images"""
         model_id = IMAGE_MODELS[model_name]
@@ -112,6 +136,7 @@ class PresentationGenerator:
                 torch_dtype=torch.bfloat16
             )
             self.image_pipeline.enable_model_cpu_offload()  # Économise de la VRAM en déchargeant le modèle sur le CPU
+            self.image_pipeline.tokenizer.add_prefix_space = False  # Désactive add_prefix_space
             print(f"Modèle d'image FLUX chargé : {model_id}")
         else:
             self.image_pipeline = pipeline(
@@ -119,7 +144,13 @@ class PresentationGenerator:
                 model=model_id,
                 token=self.token
             )
-        print(f"Modèle d'image chargé : {model_id}")
+         print(f"Modèle d'image chargé : {model_id}")
+
+
+
+
+
+
 
 
     def generate_text(self, prompt, temperature=0.7, max_tokens=4096):