Update app.py

app.py

@@ -6,14 +6,18 @@ import random
 import time
 from PIL import Image
 from diffusers import DiffusionPipeline
+import os
+
+# Set your Hugging Face token (make sure you've set this in your environment)
+hf_token = os.getenv('waffles')
 
 # Load LoRAs from JSON file
 with open('loras.json', 'r') as f:
     loras = json.load(f)
 
-# Initialize the base model
+# Initialize the base model with authentication
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16, use_auth_token=hf_token)
 
 MAX_SEED = 2**32-1