.

Dockerfile

@@ -72,8 +72,7 @@ RUN python3 -m pip install hdbscan
 RUN python3 -c "\
 from huggingface_hub import snapshot_download; \
 snapshot_download(repo_id='gregorkrzmanc/jetclustering', local_dir='models/'); \
-snapshot_download(repo_id='gregorkrzmanc/jetclustering_demo', local_dir='demo_datasets/',  \
-    ='dataset')"
+snapshot_download(repo_id='gregorkrzmanc/jetclustering_demo', local_dir='demo_datasets/', repo_type='dataset')"
 # remove pip cache
 RUN python3 -m pip cache purge