Pass HF_TOKEN explicitly to list_repo_files and hf_hub_download

backend/appfunc.py

@@ -4,11 +4,10 @@ import traceback
 import gradio as gr
 import os.path as osp
 
-from huggingface_hub import hf_hub_download, list_repo_files, login
+from huggingface_hub import hf_hub_download, list_repo_files
 
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
-if HF_TOKEN:
-    login(token=HF_TOKEN)
+print(f"HF_TOKEN present: {HF_TOKEN is not None}")
 
 from omegaconf import OmegaConf
 from refnet.util import instantiate_from_config
@@ -39,7 +38,7 @@ model_types = ["sdxl", "xlv2"]
 def get_available_models():
     """Fetch available .safetensors files from HuggingFace Hub."""
     try:
-        files = list_repo_files(HF_REPO_ID)
+        files = list_repo_files(HF_REPO_ID, token=HF_TOKEN)
         return [f for f in files if f.endswith(".safetensors")]
     except Exception as e:
         print(f"Failed to list models from {HF_REPO_ID}: {e}")
@@ -59,6 +58,7 @@ def download_model(filename):
         repo_id=HF_REPO_ID,
         filename=filename,
         local_dir=MODEL_CACHE_DIR,
+        token=HF_TOKEN,
     )
     print(f"Downloaded to {path}")
     return path