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
+
+