diff --git a/Dockerfile b/Dockerfile
index c2335d3..ea56f9b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,15 +2,15 @@ FROM ubuntu
 
 RUN apt-get update && apt-get install -y python3 python3-pip unzip
 
-RUN python3 -m pip install kaggle numpy pandas torchvision torch
- 
-COPY ium_DL.py
-COPY ./olympics-124-years-datasettill-2020/Athletes_winter_games.csv
+RUN python3 -m pip install pandas numpy tensorflow imbalanced-learn sklearn
 
-RUN chmod +r ./ium_DL.py
-RUN chmod +r ./Athletes_winter_games.csv
-RUN chmod +r ./olympics-124-years-datasettill-2020/Athletes_winter_games.csv
+COPY train.py /app/train.py
+COPY predictions.py /app/predictions.py
+COPY data.csv /app/data.csv
+COPY test_data.csv /app/test_data.csv
 
-RUN python3 ./ium_DL.py
+WORKDIR /app
 
-WORKDIR /app
\ No newline at end of file
+RUN python3 train.py
+
+RUN python3 predictions.py
\ No newline at end of file