Update app.py

app.py

@@ -10,6 +10,15 @@ import gradio as gr
 
 from src.flux.xflux_pipeline import XFluxPipeline
 
+import os
+from huggingface_hub import login
+
+hf_token = os.getenv("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
+else:
+    print("No Hugging Face token found.")
+    
 
 def list_dirs(path):
     if path is None or path == "None" or path == "":