diff --git a/Jenkinsfile_mlflow b/Jenkinsfile_mlflow
index 9c6a9d5..5c1809f 100644
--- a/Jenkinsfile_mlflow
+++ b/Jenkinsfile_mlflow
@@ -2,6 +2,7 @@ pipeline {
     agent { 
         dockerfile {
             filename 'Dockerfile'
+            args '-v /mlruns:/mlruns'
         }
     }
 
@@ -31,9 +32,8 @@ pipeline {
             steps {
                 withEnv(["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}",
                 "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
-
-
-                    sh 'rm -r mlruns'
+                    sh 'python3 ./nn_train_mlflow.py'
+                    archiveArtifacts artifacts: 'mlruns/**'
                     sh 'rm -r my_model_mlflow'
                 }
             }