nandovallec commited on
Commit
0e5008e
1 Parent(s): 23838e2
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,12 +36,12 @@ DATA_FILE_MAT = os.path.join("data_mat", DATA_FILENAME_MAT)
36
 
37
  def get_repo_train():
38
  repo_train = Repository(
39
- local_dir="data_train", clone_from=DATASET_REPO_URL_TRAIN, use_auth_token=HF_TOKEN
40
  )
41
 
42
  def get_repo_mat():
43
  repo_mat = Repository(
44
- local_dir="data_mat", clone_from=DATASET_REPO_URL_MAT, use_auth_token=HF_TOKEN
45
  )
46
 
47
  def test(playlist_url, n_rec):
 
36
 
37
  def get_repo_train():
38
  repo_train = Repository(
39
+ local_dir="data_train", clone_from=DATASET_REPO_URL_TRAIN, use_auth_token=HF_TOKEN, repo_type="dataset"
40
  )
41
 
42
  def get_repo_mat():
43
  repo_mat = Repository(
44
+ local_dir="data_mat", clone_from=DATASET_REPO_URL_MAT, use_auth_token=HF_TOKEN,repo_type="dataset"
45
  )
46
 
47
  def test(playlist_url, n_rec):