Added python dependencies

Dockerfile

@@ -2,9 +2,6 @@ FROM ubuntu:22.04
 
 WORKDIR /app
 
-COPY requirements.txt .
-RUN pip install -r requirements.txt --no-cache-dir
-
 RUN apt-get update && \
     apt-get install -y \
     git \
@@ -12,6 +9,10 @@ RUN apt-get update && \
     python3-pip && \
     rm -rf /var/lib/apt/lists/*
 
+# General python dependencies
+COPY requirements.txt .
+RUN pip3 install -r requirements.txt --no-cache-dir
+
 # Install a specific version of the transformers library
 # That implements OLMoE
 RUN git init && \
@@ -21,7 +22,7 @@ RUN git remote add origin https://github.com/huggingface/transformers.git
 RUN git fetch origin pull/32406/head:pr-32406
 RUN git checkout pr-32406
 
-RUN pip install -e .[all]
+RUN pip3 install -e .[all]
 
 COPY . .