diff --git a/s444356.py b/s444356.py
index bb08e30..686d1d2 100644
--- a/s444356.py
+++ b/s444356.py
@@ -7,6 +7,5 @@ model = mlflow.pyfunc.load_model('mlruns/1/4b83e774512444188fb587288818c298/arti
 with open('input_example.json') as file:
     data = json.load(file)
     input_example = np.array([data['inputs'][0]]).reshape(-1, 2)
-    print('Predictions:')
-    for pred in model.predict(input_example):
-        print(pred)
+    print('Input: ', input_example)
+    print('Predictions: ', model.predict(input_example))