Update requirements.txt

requirements.txt

@@ -1,16 +1,14 @@
 # ParticleNet — GNN Particle Collision Classifier
 # Hugging Face Spaces requirements
 # Python 3.10+
-#
-# Install order matters on HF Spaces:
-# torch must come before torch_geometric and its companions.
 
 # ── core deep learning ──
 torch==2.2.2
-torchvision==0.17.2   # pulled in by torch ecosystem, keeps versions aligned
+torchvision==0.17.2
 
 # ── graph neural networks ──
-# PyTorch Geometric and its sparse/scatter backends
+# Tell pip to fetch pre-built wheels for CPU instead of compiling from source
+--find-links https://data.pyg.org/whl/torch-2.2.0+cpu.html
 torch-geometric==2.5.3
 torch-scatter==2.1.2
 torch-sparse==0.6.18
@@ -18,10 +16,10 @@ torch-cluster==1.6.3
 
 # ── scientific computing ──
 numpy==1.26.4
-scikit-learn==1.4.2    # used for kNN graph construction in the jet generator
+scikit-learn==1.4.2
 
 # ── visualisation ──
-plotly==5.22.0         # interactive jet graph + training curves
+plotly==5.22.0
 
 # ── Gradio UI ──
-gradio==4.36.1
+gradio==4.36.1