diff --git a/mlflow_prediction_registry.py b/mlflow_prediction_registry.py
index 25b700e..bcee87b 100644
--- a/mlflow_prediction_registry.py
+++ b/mlflow_prediction_registry.py
@@ -18,9 +18,9 @@ model_name = "s426206"
 
 input = str(arguments[0])
 
-model = mlflow.pyfunc.load_model(
-    model_uri=f"models:/{model_name}/{model_version}"
-)
+experiment = client.get_latest_versions(model_name, stages=None)
+
+print(experiment)
 
 with open(f'{model_name}/{input}', 'r') as file:
     json_data = json.load(file)