Update app.py

app.py

@@ -2,9 +2,11 @@ import torch
 from PIL import Image
 import gradio as gr
 from huggingface_hub import hf_hub_download
+from huggingface_hub import HfFolder
 
 from src_inference.pipeline import FluxPipeline
 from src_inference.lora_helper import set_single_lora, clear_cache
+import os
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print("Running on:", device)
@@ -47,8 +49,10 @@ lora_path = hf_hub_download(
     local_dir="./LoRAs"
 )
 
+token = os.environ.get("HF_TOKEN")
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev", 
+    use_auth_token=token,
     torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32
 ).to(device)