diff --git a/Dockerfile b/Dockerfile
index c7e9272..892911a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,12 +17,15 @@ WORKDIR /app
 
 COPY ./requirements.txt ./
 COPY ./avocado-preprocessing.py ./
+COPY ./avocado-training.py ./
 
 RUN chmod +x ./requirements.txt
 RUN chmod +x ./avocado-preprocessing.py 
+RUN chmod +x ./avocado-training.py
 
 RUN pip3 install -r ./requirements.txt
 CMD python3 avocado-preprocessing.py
+CMD python3 avocado-training.py
 
 # COPY ./avocado-preprocessing.sh ./
 # RUN chmod +x avocado-preprocessing.sh