added dataset download here

utils.py

@@ -10,6 +10,7 @@ from huggingface_hub import CommitScheduler
 from functools import partial
 import pandas as pd
 import numpy as np
+from huggingface_hub import snapshot_download
 
 def enable_buttons_side_by_side():
     return tuple(gr.update(visible=True, interactive=True) for i in range(6))
@@ -29,6 +30,11 @@ no_change_btn = gr.update(value="No Change", interactive=True, visible=True)
 
 DS_ID = os.getenv('DS_ID')
 TOKEN = os.getenv('TOKEN')
+SONG_SOURCE = os.getenv("SONG_SOURCE")
+
+snapshot_download(repo_id=SONG_SOURCE, repo_type="dataset", token = TOKEN, local_dir = '/data/')
+print(os.listdir('/data/'))
+
 
 scheduler = CommitScheduler(
     repo_id= DS_ID,