Update Dockerfile

Dockerfile

@@ -8,6 +8,7 @@ COPY . .
 
 # install dependcies
 RUN conda install -y pandas numpy scikit-learn
+RUN pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.4.0+cpu.html
 RUN pip install --no-cache-dir -r requirements.txt
 
 #if you need to download executable and run them switch to the default non-root user