pass HF_TOKEN to hf_hub_download

app.py

@@ -15,6 +15,7 @@ except ImportError:
                 func = args[0]
                 return func
 
+import os
 import tempfile
 import threading
 from pathlib import Path
@@ -25,6 +26,8 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 from safetensors import torch as sft
 
+HF_TOKEN = os.environ.get("HF_TOKEN")
+
 import hyperparameters as hp
 from conditioning.condition_type import ConditionType
 from conditioning.conditioning_method import ConditioningMethod
@@ -82,17 +85,17 @@ def load_models():
     global drums_model, bass_model, model_loading, model_error
     try:
         print("Downloading shared weights...")
-        encodec_path = hf_hub_download(REPO_ID, "encodec_32khz.pt")
-        lm_path = hf_hub_download(REPO_ID, "lm-small-weights.pt")
+        encodec_path = hf_hub_download(REPO_ID, "encodec_32khz.pt", token=HF_TOKEN)
+        lm_path = hf_hub_download(REPO_ID, "lm-small-weights.pt", token=HF_TOKEN)
         params = build_params(encodec_path, lm_path)
 
         print("Building drums model (CPU)...")
-        drums_ckp = hf_hub_download(REPO_ID, "stage-drums.safetensors")
+        drums_ckp = hf_hub_download(REPO_ID, "stage-drums.safetensors", token=HF_TOKEN)
         drums_model = load_checkpoint(params, drums_ckp)
         print("Drums model ready.")
 
         print("Building bass model (CPU)...")
-        bass_ckp = hf_hub_download(REPO_ID, "stage-bass.safetensors")
+        bass_ckp = hf_hub_download(REPO_ID, "stage-bass.safetensors", token=HF_TOKEN)
         bass_model = load_checkpoint(params, bass_ckp)
         print("Bass model ready.")
     except Exception as e: