Update app.py

app.py

@@ -11,15 +11,16 @@ from diffusers.utils import load_image
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
-pipe_prior_redux = FluxPriorReduxPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-Redux-dev",
-    torch_dtype=torch.bfloat16
-).to("cuda")
-
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev" , 
-    text_encoder=None,
-    text_encoder_2=None,
+    torch_dtype=torch.bfloat16
+).to("cuda")
+pipe_prior_redux = FluxPriorReduxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-Redux-dev",
+    text_encoder=pipe.text_encoder,
+    tokenizer=pipe.tokenizer,
+    text_encoder_2=pipe.text_encoder_2,
+    tokenizer_2=pipe.tokenizer_2,
     torch_dtype=torch.bfloat16
 ).to("cuda")