Update app.py

app.py

@@ -164,13 +164,14 @@ def enhance_image(
             dtype = torch.bfloat16 if device == "cuda" else torch.float32
 
             print(f"📥 Loading FLUX Img2Img on {device}...")
-            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="tokenizer_2")
+            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="tokenizer_2", use_auth_token=huggingface_token)
             pipe = FluxImg2ImgPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
                 device_map="balanced",
-                tokenizer_2=tokenizer_2
+                tokenizer_2=tokenizer_2,
+                use_auth_token=huggingface_token
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()
@@ -196,13 +197,14 @@ def enhance_image(
             device = "cpu"
             dtype = torch.float32
             # Reload on CPU if needed
-            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="tokenizer_2")
+            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="tokenizer_2", use_auth_token=huggingface_token)
             pipe = FluxImg2ImgPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
                 device_map=None,
-                tokenizer_2=tokenizer_2
+                tokenizer_2=tokenizer_2,
+                use_auth_token=huggingface_token
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()