Update redux.py

redux.py

@@ -10,17 +10,20 @@ from diffusers.utils import load_image
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
+huggingface_token = os.getenv("HUGGINFACE_TOKEN")
 
 pipe_prior_redux = FluxPriorReduxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-Redux-dev",
-    torch_dtype=torch.bfloat16
+    torch_dtype=torch.bfloat16,
+    token=huggingface_token
 ).to("cuda")
 
 pipe = FluxPipeline.from_pretrained(
     "LPX55/FLUX.1-merged_uncensored" , 
     text_encoder=None,
     text_encoder_2=None,
-    torch_dtype=torch.bfloat16
+    torch_dtype=torch.bfloat16,
+    token=huggingface_token
 ).to("cuda")
 
 @spaces.GPU