Update app.py

app.py

@@ -2,12 +2,11 @@ import torch
 import gradio as gr
 from diffusers import Flux2Pipeline
 
+# Charger FLUX.2 COMPLET
 pipe = Flux2Pipeline.from_pretrained(
     "black-forest-labs/FLUX.2-klein-4B",
-    vae=None,
-    scheduler=None,
-    safety_checker=None,
-    feature_extractor=None,
+    torch_dtype=torch.float32,
+    low_cpu_mem_usage=True,
 )
 
 tokenizer = pipe.tokenizer