:

Dockerfile

@@ -10,6 +10,13 @@ SHELL ["/bin/bash", "-c"]
 
 USER root
 
+RUN ls /app
+RUN echo "---"
+RUN ls /app/src
+RUN echo "----"
+RUN ls /app/src/models/
+RUN echo "----"
+RUN ls /app/src/models/lgatr
 RUN apt update && \
     DEBIAN_FRONTEND=noninteractive apt install --yes --no-install-recommends \
         build-essential \
@@ -66,7 +73,8 @@ 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')"
+snapshot_download(repo_id='gregorkrzmanc/jetclustering_demo', local_dir='demo_datasets/',  \
+    ='dataset')"
 # remove pip cache
 RUN python3 -m pip cache purge
 
@@ -92,4 +100,3 @@ EXPOSE 7860
 ENV GRADIO_SERVER_NAME="0.0.0.0"
 CMD ["python", "app.py"]
 
-

app.py

@@ -23,9 +23,8 @@ 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/")
+#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")
 
 # === Interface layout ===
@@ -38,12 +37,12 @@ def gradio_ui():
             train_dataset_dropdown = gr.Dropdown(choices=["QCD", "900_03", "900_03+700_07", "700_07", "900_03+700_07+QCD"], label="Training Dataset", value="QCD")
 
         with gr.Row():
-            subdataset_dropdown = gr.Dropdown(choices=os.listdir("demo_datasets"), label="Subdataset")
+            subdataset_dropdown = gr.Dropdown(choices=[x for x in os.listdir("demo_datasets") if not x.startswith(".")], label="Subdataset")
             event_idx_dropdown = gr.Dropdown(choices=list(range(50)), label="Event Index")
         prefill_btn = gr.Button("Load Event from Dataset")
 
         particles_text = gr.Textbox(label="Particles CSV (pt eta phi mass charge)", lines=6, interactive=True)
-        quarks_text = gr.Textbox(label="Quarks CSV (pt eta phi)", lines=3, interactive=True)
+        quarks_text = gr.Textbox(label="Quarks CSV (pt eta phi) - optional", lines=3, interactive=True)
 
         process_btn = gr.Button("Run Jet Clustering")