From 9413b985c5c5013d1fa6c23963355078b75697a5 Mon Sep 17 00:00:00 2001
From: Alicja Szulecka <73056579+AliSzu@users.noreply.github.com>
Date: Sun, 14 Apr 2024 13:03:36 +0200
Subject: [PATCH] Update Jenkinsfile

---
 Jenkinsfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/Jenkinsfile b/Jenkinsfile
index 8ac9773..44578e6 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -38,12 +38,14 @@ pipeline {
 
          stage('Train and Predict') {
             steps {
+              script {
                def customImage = docker.build("custom-image")
                     customImage.inside {
                         sh 'python3 ./model.py'
                         sh 'python3 ./prediction.py'
                         archiveArtifacts artifacts: 'model.pth, predictions.txt', onlyIfSuccessful: true
                     }
+              }
             }
         }
      }