diff --git a/Jenkinsfile_predict b/Jenkinsfile_predict
index e9dec85..69f9464 100644
--- a/Jenkinsfile_predict
+++ b/Jenkinsfile_predict
@@ -16,7 +16,7 @@ pipeline {
     stages {
         stage('Script') {
             steps {
-                copyArtifacts projectName: 's444409-training/main', selector: lastSuccessful()
+                copyArtifacts projectName: 's444356-training/main', selector: lastSuccessful()
 		sh 'python3 ./predict.py $INPUT'
             }
         }
diff --git a/predict.py b/predict.py
index c523da1..f6441c0 100644
--- a/predict.py
+++ b/predict.py
@@ -1,14 +1,29 @@
+#import json
+#import mlflow
+#import sys
+
+#input = sys.argv[1]
+
+#logged_model = 'mlruns/1/3630068c31924c05a9a04e70ef35e0b8/artifacts/s444409'
+#loaded_model = mlflow.pyfunc.load_model(logged_model)
+
+
+#with open(f'{logged_model}/'+input) as f:
+#    data = json.load(f)
+
+#loaded_model.predict(data['inputs'])
+
 import json
 import mlflow
-import sys
+import numpy as np
 
-input = sys.argv[1]
-
-logged_model = 'mlruns/1/3630068c31924c05a9a04e70ef35e0b8/artifacts/s444409'
+logged_model = 'mlruns/1/296d6f314bb2451885fb7ae58988301e/artifacts/model'
 loaded_model = mlflow.pyfunc.load_model(logged_model)
 
 
 with open(f'{logged_model}/'+input) as f:
     data = json.load(f)
+    input_example = np.array([data['inputs'][0]], dtype=np.float64).reshape(-1, 2)
 
-loaded_model.predict(data['inputs'])
\ No newline at end of file
+
+print(f'Prediction: {loaded_model.predict(input_example)}')
\ No newline at end of file