Update Dockerfile

Dockerfile

@@ -1,4 +1,3 @@
-FROM ubuntu:22.04
 FROM python:3.9
 
 RUN useradd -m -u 1000 user
@@ -11,12 +10,12 @@ RUN apt-get update
 RUN apt-get install wget 
 RUN apt install -y git-all 
 RUN apt-get install -y dssp
-RUN pip install --upgrade pip
-RUN pip install git+https://github.com/a-r-j/graphein.git
-RUN pip install numpy scipy torch==2.2 transformers==4.44.2
-RUN pip install torch-geometric
-RUN pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cpu.html
-RUN pip install gradio
+RUN pip3 install --upgrade pip
+RUN pip3 install git+https://github.com/a-r-j/graphein.git
+RUN pip3 install numpy scipy torch==2.2 transformers==4.44.2
+RUN pip3 install torch-geometric
+RUN pip3 install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.2.0+cpu.html
+RUN pip3 install gradio
 
 
 COPY --chown=user . /app