From c68c84cc42368f86eb80c5504c1606774ad09a22 Mon Sep 17 00:00:00 2001
From: Witold Woch <witwoc@st.amu.edu.pl>
Date: Fri, 12 May 2023 18:41:25 +0200
Subject: [PATCH] fix

---
 Dockerfile | 32 +++++++++++++++++---------------
 1 file changed, 17 insertions(+), 15 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 7437bbd..b729512 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,18 +1,20 @@
-FROM ubuntu:latest
-
-RUN apt update && apt install -y figlet
-RUN apt install unzip -y
-RUN apt install python3-pip -y
-RUN pip3 install pandas
-RUN pip3 install kaggle
-RUN pip3 install scikit-learn
-RUN pip3 install seaborn
-
-RUN ln -s ~/.local/bin/kaggle /usr/bin/kaggle
-ENV KAGGLE_USERNAME=witeks1833
-ENV KAGGLE_KEY=""
+FROM python:3.8-slim-buster
 
 WORKDIR /app
-COPY ./script ./
-CMD ["bash", "script"]
+
+RUN apt-get update && apt-get install -y \
+    figlet \
+    unzip \
+    && rm -rf /var/lib/apt/lists/*
+RUN pip install --upgrade pip
+RUN pip install torch pandas numpy scikit-learn kaggle
+
+COPY . /app
+COPY ./train.py ./
+COPY ./prediction.py ./
+COPY ./Sales.csv ./
+
+CMD python train.py
+
+