Update Dockerfile

Dockerfile

@@ -8,6 +8,9 @@ RUN useradd -m -u 1000 user
 # Copy the requirements file to the /code directory
 COPY ./requirements.txt /code/requirements.txt
 
+# Install PyTorch
+RUN pip install torch torchvision torchaudio
+
 # Install the Python dependencies with root and clean up cache
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt && rm -rf /root/.cache