diff --git a/Jenkisfile_train b/Jenkisfile_train
index 4669e23..1df2a07 100644
--- a/Jenkisfile_train
+++ b/Jenkisfile_train
@@ -1,5 +1,10 @@
 pipeline {
-    agent {dockerfile true}
+    agent {
+        docker {
+            image 's434804/ium:0.5'
+            args '-v /tmp/mlruns:/tmp/mlruns -v /mlruns:/mlruns '
+        }
+    }
 	parameters {
 		buildSelector(
             defaultSelector: lastSuccessful(),