Update app.py

app.py

@@ -164,14 +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", use_auth_token=huggingface_token)
+            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-Krea-dev", subfolder="tokenizer_2", token=huggingface_token)
             pipe = FluxImg2ImgPipeline.from_pretrained(
-                "black-forest-labs/FLUX.1-dev",
+                "black-forest-labs/FLUX.1-Krea-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
                 device_map="balanced",
                 tokenizer_2=tokenizer_2,
-                use_auth_token=huggingface_token
+                token=huggingface_token
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()
@@ -197,14 +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", use_auth_token=huggingface_token)
+            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-Krea-dev", subfolder="tokenizer_2", token=huggingface_token)
             pipe = FluxImg2ImgPipeline.from_pretrained(
-                "black-forest-labs/FLUX.1-dev",
+                "black-forest-labs/FLUX.1-Krea-dev",
                 torch_dtype=dtype,
                 low_cpu_mem_usage=True,
                 device_map=None,
                 tokenizer_2=tokenizer_2,
-                use_auth_token=huggingface_token
+                token=huggingface_token
             )
             pipe.enable_vae_tiling()
             pipe.enable_vae_slicing()