Update app.py

app.py

@@ -10,6 +10,8 @@ dtype = torch.float32
 pipe = Flux2Pipeline.from_pretrained(
     "black-forest-labs/FLUX.2-klein-4B",
     torch_dtype=dtype,
+    text_encoder =None, 
+    vae=None, 
     low_cpu_mem_usage=True,
 )