Update app.py

app.py

@@ -10,10 +10,11 @@ import os
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if device == "cuda" else torch.float32
 
+# Required for gated access model
+auth_token = os.getenv("HUGGINGFACE_TOKEN")
+
 # Load FLUX (Kontext)
-from huggingface_hub import HfFolder
 
-auth_token = HfFolder.get_token()
 
 pipe = DiffusionPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-Kontext-dev",