Update app.py

app.py

@@ -1,12 +1,16 @@
 import torch
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM
-
-# Le vrai encodeur texte FLUX (Mistral-3-Small)
-MODEL_ID = "black-forest-labs/FLUX.2-dev/text_encoder"
+from diffusers import Flux2Pipeline
+import transformers 
+pipe = Flux2Pipeline.from_pretrained(
+    "black-forest-labs/FLUX.2-klein-4B",
+    vae=None,
+    scheduler=None,
+    safety_checker=None,
+)
 
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
-text_encoder = AutoModelForCausalLM.from_pretrained(MODEL_ID)
+tokenizer = pipe.tokenizer
+text_encoder = pipe.text_encoder
 
 def encode_text(prompt: str):
     inputs = tokenizer(