Upload docker_start.py with huggingface_hub

docker_start.py

@@ -61,13 +61,16 @@ def main() -> None:
 
  # Download the huggingface space data. This includes code and datasets, so we move the datasets
  # alone to the data directory.
+
+ datasets_dir = get_datasets_dir(get_project_dir())
+ os.makedirs(datasets_dir, exist_ok=True)
  for lilac_hf_dataset in hf_config['datasets']:
  print('Downloading dataset from HuggingFace: ', lilac_hf_dataset)
  snapshot_download(
  repo_id=lilac_hf_dataset,
  repo_type='dataset',
  token=env('HF_ACCESS_TOKEN'),
- local_dir=get_datasets_dir(get_project_dir()),
+ local_dir=datasets_dir,
  ignore_patterns=['.gitattributes', 'README.md'])
 
  snapshot_dir = snapshot_download(repo_id=repo_id, repo_type='space', token=env('HF_ACCESS_TOKEN'))