Update Dockerfile

Dockerfile

@@ -10,13 +10,17 @@ RUN apt-get update && apt-get upgrade -y && apt-get install -y --no-install-reco
     gnupg2 \
     wget
 
+
+
 WORKDIR /code
 
 RUN chmod 777 .
 
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
 COPY ./requirements.txt /code/requirements.txt
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
 
 RUN useradd -m -u 1000 user