diff --git a/JenkinsfileDL b/JenkinsfileDL
index 3f5a32a..4eab5ab 100644
--- a/JenkinsfileDL
+++ b/JenkinsfileDL
@@ -3,10 +3,7 @@ pipeline {
         docker {
             image 'python:3.11'
             args '-v /root/.cache:/root/.cache'
-            script {
-                // sh 'python3 train.py ${params.EPOCHS}'
-                sh 'pip install pandas && tail -f /dev/null'
-            }
+            command 'sh', '-c', 'pip install pandas tensorflow && tail -f /dev/null'
         }
     }
     parameters {
@@ -23,8 +20,7 @@ pipeline {
         stage('Trenuj model') {
             steps {
                 script {
-                    // sh 'python3 train.py ${params.EPOCHS}'
-                    sh 'python3 train.py'
+                    sh 'python3 train.py --epochs ${params.EPOCHS}'
                 }
             }
         }
@@ -34,4 +30,4 @@ pipeline {
             }
         }
     }
-}
\ No newline at end of file
+}