login to huggingface

app.py

@@ -4,10 +4,14 @@ import time
 import torch
 import gradio as gr
 from PIL import Image
-from huggingface_hub import hf_hub_download, list_repo_files
+from huggingface_hub import hf_hub_download, list_repo_files, login
 from src_inference.pipeline import FluxPipeline
 from src_inference.lora_helper import set_single_lora
 
+HF_TOKEN = os.environ.get("HF_TOKEN") 
+if HF_TOKEN:
+    login(token=HF_TOKEN)
+
 BASE_PATH = "black-forest-labs/FLUX.1-dev"
 LOCAL_LORA_DIR = "./LoRAs"          
 CUSTOM_LORA_DIR = "./Custom_LoRAs"