How to enable and disable Model Serving using Rest API

[Mohit_m]

ML flow model serving in Databricks docs details the options to enable and disable from the UI

https://docs.databricks.com/applications/mlflow/model-serving.html

[Mohit_m · Accepted solution]

Please find below the REST APIs to enable and disable Model-Serving

Below are the examples in Python

You need to use the token to interact with Rest API

token = "dxxxxxx"

instance = "https://<workspacexxx>.cloud.databricks.com"

headers = {'Authorization': f'Bearer {token}'}

Enable Model Serving

import requests

url = f'https://{instance}/api/2.0/mlflow/endpoints/enable'

requests.post(url, headers=headers, json={"registered_model_name": "<model name>"})

Disable Model serving

url = f'https://{instance}/api/2.0/mlflow/endpoints/disable'

r = requests.post(url, headers=headers, json={"registered_model_name": "<model name>"})

r.json()

[jose_gonzalez]

Hi @Mohit Miglani​ ,

Great post. Thank you for sharing it with the rest of the community.