Update app.py

app.py

@@ -133,6 +133,7 @@ class PresentationGenerator:
         if model_id == "black-forest-labs/FLUX.1-schnell":
             self.image_pipeline = FluxPipeline.from_pretrained(
                 model_id,
+                revision="refs/pr/1",  # Utiliser une révision spécifique
                 torch_dtype=torch.bfloat16
             )
             self.image_pipeline.enable_model_cpu_offload()  # Économise de la VRAM en déchargeant le modèle sur le CPU