diff --git a/Rest_API/Dockerfile b/Rest_API/Dockerfile
index 9987fcb..b25ab94 100644
--- a/Rest_API/Dockerfile
+++ b/Rest_API/Dockerfile
@@ -1,7 +1,5 @@
 FROM python:latest
 
-RUN pip install -r requirements.txt
-
 WORKDIR /app
 COPY . /app
 
@@ -20,6 +18,8 @@ RUN apt-get update && \
     apt-get install -y wget && \
     rm -rf /var/lib/apt/lists/
 
+RUN pip install -r requirements.txt
+
 USER appuser
 
 EXPOSE 32196
diff --git a/Train_model/Dockerfile b/Train_model/Dockerfile
index ecf6f64..23f24ce 100644
--- a/Train_model/Dockerfile
+++ b/Train_model/Dockerfile
@@ -1,7 +1,5 @@
 FROM python:latest
 
-RUN pip install -r requirements.txt
-
 WORKDIR /app
 COPY . /app
 
@@ -16,6 +14,8 @@ RUN adduser \
     --uid "${UID}" \
     appuser
 
+RUN pip install -r requirements.txt
+
 USER root
 
 EXPOSE 32208