diff --git a/predict-registry/predict.py b/predict-registry/predict.py index 22e22c5..6c9b2f9 100644 --- a/predict-registry/predict.py +++ b/predict-registry/predict.py @@ -14,4 +14,4 @@ model = mlflow.keras.pyfunc.load_model( ) client = MlflowClient() -print(client.get_model_version(model_name, model_version).source) +print("XD: ", client.get_model_version(model_name, model_version).source)