diff --git a/Dockerfile b/Dockerfile
index c331eb1..e48d2a9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,4 +17,6 @@ WORKDIR /app
 
 COPY . /app
 
-CMD ["python", "create_dataset.py"]
\ No newline at end of file
+CMD ["python", "create_dataset.py"]
+CMD ["python", "train.py"]
+CMD ["python", "pred.py"]
\ No newline at end of file