Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import spaces  # Import this first to avoid CUDA initialization issues
 import gradio as gr
 import json
@@ -6,9 +7,8 @@ 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)
+# Use the 'waffles' environment variable as the access token
 hf_token = os.getenv('waffles')
 
 # Load LoRAs from JSON file
@@ -17,7 +17,11 @@ with open('loras.json', 'r') as f:
 
 # Initialize the base model with authentication
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16, use_auth_token=hf_token)
+pipe = DiffusionPipeline.from_pretrained(
+    base_model,
+    torch_dtype=torch.bfloat16,
+    use_auth_token=hf_token  # Use the 'waffles' token
+)
 
 MAX_SEED = 2**32-1