.

Dockerfile

@@ -66,7 +66,7 @@ RUN python3 -m pip install fastjet
 RUN python3 -m pip install gradio
 RUN python3 -m pip install huggingface_hub
 RUN python3 -m pip install hdbscan
-
+RUN poython3 -m pip install lgatr
 RUN python3 -c "\
 from huggingface_hub import snapshot_download; \
 snapshot_download(repo_id='gregorkrzmanc/jetclustering', local_dir='models/'); \
@@ -82,9 +82,10 @@ RUN mkdir -p /opt/pepr
 #RUN cd /opt/pepr && git clone https://github.com/Qualcomm-AI-research/geometric-algebra-transformer.git geometric-algebra-transformer1
 #RUN cd /opt/pepr/geometric-algebra-transformer1/ && python3 -m pip install .
 
+
 # Install L-GATr - for some reason this only works if executed from the already-built container
-RUN cd /opt/pepr && git clone https://github.com/gregorkrz/lorentz-gatr lgatr
-RUN cd /opt/pepr/lgatr/ && python3 -m pip install .
+#RUN cd /opt/pepr && git clone https://github.com/gregorkrz/lorentz-gatr lgatr
+#RUN cd /opt/pepr/lgatr/ && python3 -m pip install .
 
 # Install torch_cmspepr