Update app.py

app.py

@@ -164,7 +164,7 @@ 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="text_encoder_2")
+            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="tokenizer_2")
             pipe = FluxImg2ImgPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,
@@ -196,7 +196,7 @@ 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="text_encoder_2")
+            tokenizer_2 = T5TokenizerFast.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="tokenizer_2")
             pipe = FluxImg2ImgPipeline.from_pretrained(
                 "black-forest-labs/FLUX.1-dev",
                 torch_dtype=dtype,