diff --git a/doc/source/serve/ml-models.rst b/doc/source/serve/ml-models.rst index dc1ad4c56..8fe3330af 100644 --- a/doc/source/serve/ml-models.rst +++ b/doc/source/serve/ml-models.rst @@ -90,7 +90,8 @@ class will allow you to load a model using its MLflow `Model URI`: df = pd.read_csv(csv_text) return self.model.predict(df) - MLflowDeployment.deploy() + model_uri = "model:/my_registered_model/Production" + MLflowDeployment.deploy(model_uri) .. tip::