use hf hub to login instead of passing the token

main.py

@@ -15,13 +15,15 @@ from audiocraft.models import JASCO
 import os
 from huggingface_hub import login
 
+hf_token = os.environ.get('HFTOKEN')
+if hf_token:
+    login(token=hf_token)
+
 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
@@ -69,13 +71,12 @@ 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, token=hf_token)
+        MODEL = JASCO.get_pretrained(version)
 
 @spaces.GPU
 def _do_predictions(texts, chords, melody_matrix, drum_prompt, progress=False, gradio_progress=None, **gen_kwargs):