diff --git a/Jenkinsfile-training b/Jenkinsfile-training
index 72b62ff..53372e8 100644
--- a/Jenkinsfile-training
+++ b/Jenkinsfile-training
@@ -19,8 +19,7 @@ pipeline{
 
         stage('Train model'){
             steps{
-                def epochs = params.EPOCHS ?: '10' // Domyślna wartość 10, jeśli parametr EPOCHS nie jest przekazany
-                sh "docker run -v ${env.WORKSPACE}/dane:/app/dane ium python3 /app/DL-model.py --epochs ${epochs}"
+                sh "docker run -v ${env.WORKSPACE}/dane:/app/dane ium python3 /app/DL-model.py --epochs ${parameters.EPOCHS}"
             }
         }