anon4757 commited on
Commit
0e54f18
·
1 Parent(s): 65bab5c

Update ds_manager.py

Browse files
Files changed (1) hide show
  1. ds_manager.py +1 -1
ds_manager.py CHANGED
@@ -8,7 +8,7 @@ import huggingface_hub
8
  print("hfh", huggingface_hub.__version__)
9
  from huggingface_hub import hf_hub_download, upload_file, delete_file, snapshot_download, list_repo_files, dataset_info
10
 
11
- DATASET_REPO_ID = "AnimaLab/vision_model_init_data"
12
  DATASET_REPO_URL = f"https://huggingface.co/{DATASET_REPO_ID}"
13
  HF_DATA_DIRNAME = "data"
14
  LOCAL_DATA_DIRNAME = "data"
 
8
  print("hfh", huggingface_hub.__version__)
9
  from huggingface_hub import hf_hub_download, upload_file, delete_file, snapshot_download, list_repo_files, dataset_info
10
 
11
+ DATASET_REPO_ID = os.environ.get("DATASET_REPO_ID")
12
  DATASET_REPO_URL = f"https://huggingface.co/{DATASET_REPO_ID}"
13
  HF_DATA_DIRNAME = "data"
14
  LOCAL_DATA_DIRNAME = "data"