.

Dockerfile

@@ -63,7 +63,10 @@ RUN python3 -m pip install gradio
 RUN python3 -m pip install huggingface_hub
 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/', repo_type='dataset')"
 # remove pip cache
 RUN python3 -m pip cache purge
 

app.py

@@ -23,10 +23,10 @@ def prefill_event(subdataset, event_idx):
     return particles_data, quarks_data
 
 
-from huggingface_hub import snapshot_download
-
-snapshot_download(repo_id="gregorkrzmanc/jetclustering", local_dir="src/models/")
-snapshot_download(repo_id="gregorkrzmanc/jetclustering_demo", local_dir="demo_datasets/", repo_type="dataset")
+#from huggingface_hub import snapshot_download#
+#
+#snapshot_download(repo_id="gregorkrzmanc/jetclustering", local_dir="src/models/")
+#snapshot_download(repo_id="gregorkrzmanc/jetclustering_demo", local_dir="demo_datasets/", repo_type="dataset")
 
 # === Interface layout ===
 def gradio_ui():