diff --git a/MLFLOW/Dockerfile b/MLFLOW/Dockerfile
index fb2d6d7..3935430 100644
--- a/MLFLOW/Dockerfile
+++ b/MLFLOW/Dockerfile
@@ -9,5 +9,7 @@ RUN python -m pip install -r /tmp/requirements.txt
 
 WORKDIR ./app
 COPY train.py ./
+COPY df_train.csv ./
+COPY df_test.csv ./
 
 CMD bash
\ No newline at end of file