diff --git a/Jenkinsfile-training b/Jenkinsfile-training
index ad1e905..08c8357 100644
--- a/Jenkinsfile-training
+++ b/Jenkinsfile-training
@@ -11,7 +11,8 @@ pipeline{
 
         stage('Train model'){
             steps{
-                sh "docker run -v ${env.WORKSPACE}/dane:app/dane ium python3 /app/DL-model.py"
+                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}"
             }
         }