diff --git a/Jenkinsfile b/Jenkinsfile
index cdbc456..fa3899c 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -43,7 +43,7 @@ pipeline {
             steps {
                 sh "chmod +x ./IUM_05-model.py"
                 sh "chmod +x ./IUM_05-predict.py"
-                sh "python3 ./IUM_05-model.py"
+                sh "python3 ./IUM_05-model.py 10 32"
                 sh "python3 ./IUM_05-predict.py"
                 archiveArtifacts artifacts: 'beer_review_sentiment_model.h5,beer_review_sentiment_predictions.csv', onlyIfSuccessful: true
             }