Fix

Dockerfile

@@ -1,6 +1,7 @@
 FROM ubuntu:latest
 
 RUN apt update && apt install -y figlet
+RUN apt install unzip -y
 RUN apt install python3-pip -y
 
 RUN pip3 install kaggle

Jenkinsfile_train

@@ -10,7 +10,7 @@ node {
      checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's487197', url: 'https://git.wmi.amu.edu.pl/s487197/ium_487197']]])
  }
  stage('Dockerfile'){
-    def testImage = docker.image('s487197/ium:33')
+    def testImage = docker.image('s487197/ium:34')
 	testImage.inside{ 
          copyArtifacts filter: 'baltimore_train.csv', projectName: 's487197-create-dataset'
 		 sh "python3 ium_train.py"

ium_predict.py

@@ -26,7 +26,7 @@ def get_x_y(data):
 
 
 def predict():
-    model = load_model('baltimore_model3')
+    model = load_model('baltimore_model.h5')
 
     train = pd.read_csv('baltimore_train.csv')
     baltimore_data_test = pd.read_csv('baltimore_test.csv')

ium_train.py

@@ -48,7 +48,7 @@ def train_model():
         validation_split=0.2)
     hist = pd.DataFrame(history.history)
     hist['epoch'] = history.epoch
-    model.save('baltimore_model3')
+    model.save('baltimore_model.h5')
 
 
 train_model()