Update app.py

app.py

@@ -28,6 +28,9 @@ os.environ["HF_HOME"] = os.path.join(os.getcwd(), ".cache/huggingface")
 import torch
 from diffusers import FluxFillPipeline
 
+import warnings
+warnings.filterwarnings("ignore", message=".*add_prefix_space.*")
+
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -95,11 +98,12 @@ def load_model():
     st.info(f"Token available: {'Yes' if token else 'No'}")
     
     try:
-        # Use the same parameters as the Gradio version, just with token
+        # Add use_fast_tokenizer=True to address the tokenizer warning
         model = FluxFillPipeline.from_pretrained(
             "black-forest-labs/FLUX.1-Fill-dev",
             token=token,
-            torch_dtype=torch.bfloat16
+            torch_dtype=torch.bfloat16,
+            tokenizer_config={"add_prefix_space": False}
         )
         st.success("Model loaded successfully!")
         return model.to(device)