diff --git a/assets/training/model_management/environments/model-management/context/Dockerfile b/assets/training/model_management/environments/model-management/context/Dockerfile
index eb1475358b..536d468bde 100644
--- a/assets/training/model_management/environments/model-management/context/Dockerfile
+++ b/assets/training/model_management/environments/model-management/context/Dockerfile
@@ -4,29 +4,29 @@ WORKDIR /
 
 RUN apt-get update && apt-get upgrade -y && apt-get install wget nscd python3-idna git git-lfs libxml2 -y
 
-# azcopy install starts
-RUN echo "Downloading azcopy to file azcopy.tar ....";\
-    wget https://aka.ms/downloadazcopy-v10-linux -O azcopy.tar --no-verbose;\
-    tar -xvf azcopy.tar;\
-    mkdir -p ./azcopy/bin/;\
-    cp ./azcopy_linux_amd64_*/azcopy ./bin/;\
-    rm azcopy.tar;\
-    rm -rf ./azcopy_linux_amd64_*;\
-    which azcopy | grep -o azcopy > /dev/null &&  echo "azcopy not installed" || echo "azcopy installed";
+# Upgrade Go to 1.24.4 to fix CVE-2025-22874
+RUN wget https://go.dev/dl/go1.24.4.linux-amd64.tar.gz && \
+    tar -C /usr/local -xzf go1.24.4.linux-amd64.tar.gz && \
+    rm go1.24.4.linux-amd64.tar.gz
+
+ENV PATH="/usr/local/go/bin:$PATH"
 
 COPY requirements.txt .
 
 # This line is added because azureml-automl-dnn-vision depends on python 3.9.
 # Added ignore python requires to get this installed.
-RUN pip install --ignore-requires-python azureml-automl-dnn-vision
+# RUN pip install --ignore-requires-python azureml-automl-dnn-vision
+RUN pip install azureml-automl-dnn-vision
 
 RUN pip install -r requirements.txt --no-cache-dir
 
+RUN pip install --upgrade nltk==3.9
+
 # Adding this to exclude Python 3.13 packages from the setuptools library fix
 RUN rm -rf /opt/conda/lib/python3.13 || true
 
 # List pip packages
 RUN pip list
 
-## Delete
+# Cleanup
 RUN rm requirements.txt
diff --git a/assets/training/model_management/environments/model-management/context/requirements.txt b/assets/training/model_management/environments/model-management/context/requirements.txt
index 916548a531..d2a2b1c0c6 100644
--- a/assets/training/model_management/environments/model-management/context/requirements.txt
+++ b/assets/training/model_management/environments/model-management/context/requirements.txt
@@ -32,3 +32,4 @@ scikit-learn==1.5.1
 mlflow==3.1.0
 mlflow-skinny==3.1.0
 marshmallow==3.23.2
+transformers==4.52.1