diff --git a/Dockerfile b/Dockerfile
index 64ca3d9..175b548 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -19,7 +19,11 @@ RUN pip3 install  keras
 RUN pip3 install  GitPython
 RUN pip3 install  pymongo
 RUN pip3 install  mlflow
+RUN mkdir tmp
+RUN cd tmp
+RUN mkdir mlruns
+RUN cd ..
+RUN cd ..
+
 
-WORKDIR /app
 
-COPY ./train.py ./