Add HFTOKEN

main.py

@@ -12,11 +12,17 @@ import gradio as gr
 from audiocraft.data.audio_utils import f32_pcm, normalize_audio
 from audiocraft.data.audio import audio_write
 from audiocraft.models import JASCO
+import os
+from huggingface_hub import login
 
 MODEL = None
 MAX_BATCH_SIZE = 12
 INTERRUPTING = False
 
+hf_token = os.environ.get('HFTOKEN')
+if hf_token:
+    login(token=hf_token)
+
 # Wrap subprocess call to clean logs
 _old_call = sp.call
 
@@ -63,12 +69,13 @@ def chords_string_to_list(chords: str):
     chrd_times = [x.split(',') for x in chords[1:-1].split('),(')]
     return [(x[0], float(x[1])) for x in chrd_times]
 
+
 def load_model(version='facebook/jasco-chords-drums-400M'):
     global MODEL
     print("Loading model", version)
     if MODEL is None or MODEL.name != version:
         MODEL = None
-        MODEL = JASCO.get_pretrained(version)
+        MODEL = JASCO.get_pretrained(version, token=hf_token)
 
 @spaces.GPU
 def _do_predictions(texts, chords, melody_matrix, drum_prompt, progress=False, gradio_progress=None, **gen_kwargs):

requirements.txt

@@ -8,4 +8,5 @@ scipy
 einops
 rotary_embedding_torch
 xformers
-demucs
+demucs
+huggingface_hub