diff --git a/Dockerfile b/Dockerfile
index 6a6f4e7..db8a3ca 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,6 +3,7 @@ FROM ubuntu
 RUN apt-get update && apt-get install -y python3 python3-pip unzip
 
 RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn
+RUN apt-get install -y git
 
 COPY train.py /app/train.py
 COPY predictions.py /app/predictions.py