Update app.py

app.py

@@ -2,11 +2,22 @@ import gradio as gr
 import spaces
 import random
 
+from huggingface_hub import login
+import os
+
+
+login(token=os.environ["HUGGINGFACE_HUB_TOKEN"])
+
+
 import torch
 from diffusers import FluxPipeline
 from huggingface_hub.utils import RepositoryNotFoundError
 
-pipeline = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16).to("cuda")
+pipeline = FluxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-dev",
+    torch_dtype=torch.float16,
+    use_auth_token=True  # This uses the environment token if not passed explicitly
+).to("cuda")
 
 @spaces.GPU(duration=70)
 def generate(prompt, negative_prompt, width, height, sample_steps, lora_id):