Update space

app.py

@@ -8,11 +8,13 @@ import os
 from PIL import Image
 from diffusers import FluxKontextPipeline
 from diffusers.utils import load_image
-from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard
+from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, login
 from safetensors.torch import load_file
 import requests
 import re
 
+login(token=os.environ.get("HF_TOKEN"))
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 MAX_SEED = np.iinfo(np.int32).max