diff --git a/train_evaluate/Dockerfile b/train_evaluate/Dockerfile
index c06dfb0..bd373de 100644
--- a/train_evaluate/Dockerfile
+++ b/train_evaluate/Dockerfile
@@ -5,6 +5,7 @@ FROM ubuntu:latest
 RUN apt update && apt install -y figlet
 RUN apt install -y git
 RUN apt install -y python3-pip
+RUN pip3 install --user setuptools
 RUN pip3 install --user kaggle
 RUN pip3 install --user pandas
 RUN pip3 install --user numpy
@@ -14,6 +15,8 @@ RUN pip3 install --user matplotlib
 RUN pip3 install --user tensorflow
 RUN pip3 install --user sacred
 RUN pip3 install --user wget
+RUN pip3 install --user keras
+
 
 WORKDIR /app
 
@@ -23,5 +26,4 @@ COPY ./sacred1.py ./
 COPY ./sacred2.py ./
 COPY ./skrypt.sh ./
 COPY ./zadanie2.py ./
-COPY ./zadanie5.py ./
-
+COPY ./zadanie5.py ./
\ No newline at end of file