diff --git a/get_registered_model_via_rest_api.py b/get_registered_model_via_rest_api.py new file mode 100644 index 0000000..c5f6664 --- /dev/null +++ b/get_registered_model_via_rest_api.py @@ -0,0 +1,30 @@ +# get_registered_model_via_rest_api.py +# +# author: deng +# date : 20230224 + +import json +import requests + +def main(): + + registered_model_name = 'fortune_predict_model' + production_model_version = None + + query = {'name': registered_model_name} + res = requests.get('http://127.0.0.1:5000/api/2.0/mlflow/registered-models/get', params=query) + content = json.loads(res.text) + print(content) + + for model in content['registered_model']['latest_versions']: + + if model['current_stage'] == 'Production': + production_model_version = model['version'] + + if production_model_version is not None: + query = {'name': registered_model_name, 'version': production_model_version} + res = requests.get('http://127.0.0.1:5000/api/2.0/mlflow/model-versions/get-download-uri', params=query) + print(res.text) + +if __name__ == '__main__': + main() \ No newline at end of file diff --git a/start_mlflow_server.sh b/start_mlflow_server.sh old mode 100644 new mode 100755