diff --git a/predict.py b/predict.py
index 5fad2de..6929115 100644
--- a/predict.py
+++ b/predict.py
@@ -14,5 +14,5 @@ model =mlflow.pyfunc.load_model(f"{PATH}")
 with open(f'{PATH}{input}', 'r') as file:
     json_data = json.load(file)
 
-    print(f"Input: {json_data['inputs'][0]}")
-    print(f"Prediction: {model.predict(np.array([json_data['inputs'][0]], dtype=np.float32))}")
\ No newline at end of file
+    print(f"Input: {json_data['inputs']}")
+    print(f"Prediction: {model.predict(np.array([json_data['inputs']], dtype=np.float32))}")
\ No newline at end of file