diff --git a/predict-registry/predict.py b/predict-registry/predict.py
index dc3daa3..22e22c5 100644
--- a/predict-registry/predict.py
+++ b/predict-registry/predict.py
@@ -12,3 +12,6 @@ mlflow.set_tracking_uri("http://172.17.0.1:5000")
 model = mlflow.keras.pyfunc.load_model(
     model_uri=f"models:/{model_name}/{model_version}",
 )
+
+client = MlflowClient()
+print(client.get_model_version(model_name, model_version).source)