diff --git a/flask_ml_api/Dockerfile b/flask_ml_api/Dockerfile
index bdc1fc03587f7ede942c101878f48d7fc00adcc8..babb246855dcd5eebe462b5c44f1f10075596192 100755
--- a/flask_ml_api/Dockerfile
+++ b/flask_ml_api/Dockerfile
@@ -1,5 +1,10 @@
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime as build
 
+RUN apt-get update && apt-get install -y curl
+RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs -o rustup-init.sh
+RUN sh rustup-init.sh -y --default-toolchain none
+RUN ~/.cargo/bin/rustup toolchain install stable --profile minimal --no-self-update
+
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential python3-venv && \
     apt-get clean && \
@@ -15,7 +20,6 @@ RUN pip install --no-cache-dir Flask matplotlib scikit-learn transformers pandas
     pip install --no-cache-dir --upgrade setuptools wheel && \
     rm -rf /root/.cache/pip
 
-
 FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-runtime
 
 COPY --from=build /opt/myenv /opt/myenv