Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
 import gradio as gr
+from transformers import Qwen3ForCausalLM 
 from diffusers import Flux2Pipeline
 
 # Charger FLUX.2 Klein COMPLET pour récupérer le vrai text encoder (Qwen3)
@@ -15,7 +16,7 @@ pipe = Flux2Pipeline.from_pretrained(
 
 # Récupération du tokenizer + text_encoder (Qwen3ForCausalLM)
 tokenizer = pipe.tokenizer
-text_encoder = pipe.text_encoder
+text_encoder = Qwen3ForCausalLM.from_pretained(pipe.text_encoder)
 
 def encode_text(prompt: str):
     # Tokenisation simple