From 7a4759e8b1530e2202614b7ee0479f4299c44bff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Wojciech=20Miko=C5=82ajski?= <wojmik2@st.amu.edu.pl>
Date: Sat, 17 Jun 2023 20:10:28 +0200
Subject: [PATCH] Update 'Dockerfile'

---
 Dockerfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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