From a614dc62b9f475ce6f57dfe900daaf5144cd1e5f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Zar=C4=99ba?= <k.mjayk@gmail.com>
Date: Wed, 10 May 2023 14:11:16 +0200
Subject: [PATCH] Feat: Dockerfile fix

---
 Dockerfile | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

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