diff --git a/Jenkinsfile b/Jenkinsfile
index 356c2fd..cf44117 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,5 +1,10 @@
 pipeline {
-    agent { dockerfile true }
+    agent { 
+        dockerfile {
+        filename 'Dockerfile'
+        args '-v /tmp:/tmp'
+        }
+    }
 
    parameters{
       buildSelector(
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/meta.yaml b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/meta.yaml
new file mode 100644
index 0000000..aac2b1b
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/meta.yaml
@@ -0,0 +1,15 @@
+artifact_uri: file:///home/tomasz/IUM/ium_434695/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/artifacts
+end_time: 1621803527317
+entry_point_name: ''
+experiment_id: '0'
+lifecycle_stage: active
+name: ''
+run_id: f5c7f50e058248b0b2ccdb2a43dbac38
+run_uuid: f5c7f50e058248b0b2ccdb2a43dbac38
+source_name: ''
+source_type: 4
+source_version: ''
+start_time: 1621803499546
+status: 3
+tags: []
+user_id: tomasz
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/metrics/rmse b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/metrics/rmse
new file mode 100644
index 0000000..1d45fa4
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/metrics/rmse
@@ -0,0 +1 @@
+1621803525760 0.0427710843373494 0
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/params/batch_size b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/params/batch_size
new file mode 100644
index 0000000..19c7bdb
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/params/batch_size
@@ -0,0 +1 @@
+16
\ No newline at end of file
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/params/epochs b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/params/epochs
new file mode 100644
index 0000000..3f10ffe
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/params/epochs
@@ -0,0 +1 @@
+15
\ No newline at end of file
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.git.commit b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.git.commit
new file mode 100644
index 0000000..6e7b32a
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.git.commit
@@ -0,0 +1 @@
+eee8d50045b94596c5132c29b393ab28d589ca2c
\ No newline at end of file
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.name b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.name
new file mode 100644
index 0000000..939bfd8
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.name
@@ -0,0 +1 @@
+/home/tomasz/IUM/ium_434695/vgsales-mlflow.py
\ No newline at end of file
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.type b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.type
new file mode 100644
index 0000000..0c2c1fe
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.source.type
@@ -0,0 +1 @@
+LOCAL
\ No newline at end of file
diff --git a/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.user b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.user
new file mode 100644
index 0000000..4b82f83
--- /dev/null
+++ b/mlruns/0/f5c7f50e058248b0b2ccdb2a43dbac38/tags/mlflow.user
@@ -0,0 +1 @@
+tomasz
\ No newline at end of file
diff --git a/my_model/MLmodel b/my_model/MLmodel
index 9acac1c..aeac31c 100644
--- a/my_model/MLmodel
+++ b/my_model/MLmodel
@@ -18,4 +18,4 @@ signature:
     {"name": "JP_Sales", "type": "double"}, {"name": "Other_Sales", "type": "double"},
     {"name": "Global_Sales", "type": "double"}]'
   outputs: '[{"type": "long"}]'
-utc_time_created: '2021-05-23 17:35:24.878064'
+utc_time_created: '2021-05-23 20:58:45.764716'
diff --git a/my_model/data/model/saved_model.pb b/my_model/data/model/saved_model.pb
index 752a22f..caa52c2 100644
Binary files a/my_model/data/model/saved_model.pb and b/my_model/data/model/saved_model.pb differ
diff --git a/my_model/data/model/variables/variables.data-00000-of-00001 b/my_model/data/model/variables/variables.data-00000-of-00001
index 9ac6226..154ea93 100644
Binary files a/my_model/data/model/variables/variables.data-00000-of-00001 and b/my_model/data/model/variables/variables.data-00000-of-00001 differ
diff --git a/my_model/data/model/variables/variables.index b/my_model/data/model/variables/variables.index
index 70018ec..6f781f1 100644
Binary files a/my_model/data/model/variables/variables.index and b/my_model/data/model/variables/variables.index differ
diff --git a/predict434788.py b/predict434788.py
new file mode 100644
index 0000000..ff4b225
--- /dev/null
+++ b/predict434788.py
@@ -0,0 +1,7 @@
+import mlflow
+import pandas as pd
+
+model = mlflow.keras.load_model("my_model")
+data = pd.read_json('my_model/input_example.json', orient='split')
+
+print(model.predict(data))
\ No newline at end of file
diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py
index 81716fb..7ef1f2a 100644
--- a/vgsales-mlflow.py
+++ b/vgsales-mlflow.py
@@ -35,7 +35,7 @@ def my_main(epochs, batch_size):
     early_stop = EarlyStopping(monitor="val_loss", mode="min", verbose=1, patience=10)
 
 
-    model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
+    model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
 
 
     model.fit(X_train, y_train, epochs=epochs, batch_size=batch_size, validation_data=(X_test, y_test))
diff --git a/vgsales_model.h5 b/vgsales_model.h5
index b8dceb1..a1fa4f4 100644
Binary files a/vgsales_model.h5 and b/vgsales_model.h5 differ