diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver.py b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver.py
index 5e8d611f34..a72a794d2d 100644
--- a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver.py
+++ b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver.py
@@ -19,7 +19,7 @@ def init():
     model_path = glob.glob(os.environ["AZUREML_MODEL_DIR"] + "/*/")[0]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver_parquet.py b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver_parquet.py
index 2cf82f970c..026aa86f8f 100644
--- a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver_parquet.py
+++ b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/code/batch_driver_parquet.py
@@ -22,7 +22,7 @@ def init():
     output_path = os.environ["AZUREML_BI_OUTPUT_PATH"]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/MLmodel b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/MLmodel
deleted file mode 100644
index ec4997ca7c..0000000000
--- a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/MLmodel
+++ /dev/null
@@ -1,22 +0,0 @@
-artifact_path: pipeline
-flavors:
-  python_function:
-    env: conda.yaml
-    loader_module: mlflow.sklearn
-    model_path: model.pkl
-    python_version: 3.8.5
-  sklearn:
-    pickled_model: model.pkl
-    serialization_format: cloudpickle
-    sklearn_version: 1.1.2
-model_uuid: 04bd660a1b8b4b1e84b9198c46cfd117
-run_id: 22874b7e-b069-43f0-b90c-1c57793c7854
-signature:
-  inputs: '[{"name": "age", "type": "long"}, {"name": "sex", "type": "long"},
-    {"name": "cp", "type": "long"}, {"name": "trestbps", "type": "long"}, {"name":
-    "chol", "type": "long"}, {"name": "fbs", "type": "long"}, {"name": "restecg",
-    "type": "long"}, {"name": "thalach", "type": "long"}, {"name": "exang", "type":
-    "long"}, {"name": "oldpeak", "type": "double"}, {"name": "slope", "type": "long"},
-    {"name": "ca", "type": "long"}, {"name": "thal", "type": "string"}]'
-  outputs: '[{"type": "long"}]'
-utc_time_created: '2022-10-13 00:55:57.543663'
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/conda.yaml b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/conda.yaml
deleted file mode 100644
index 0605ffc2eb..0000000000
--- a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/conda.yaml
+++ /dev/null
@@ -1,11 +0,0 @@
-channels:
-- conda-forge
-dependencies:
-- python=3.8.5
-- pip
-- pip:
-  - mlflow
-  - cloudpickle==1.6.0
-  - scikit-learn==1.1.2
-  - xgboost==1.3.3
-name: mlflow-env
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/model.pkl b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/model.pkl
deleted file mode 100644
index febb48669e..0000000000
Binary files a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/model.pkl and /dev/null differ
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/requirements.txt b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/requirements.txt
deleted file mode 100644
index 2c6d307cc8..0000000000
--- a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-simple/model/requirements.txt
+++ /dev/null
@@ -1,15 +0,0 @@
-mlflow
-cloudpickle==1.6.0
-colorama==0.4.4
-dask==2.30.0
-dataclasses==0.6
-lz4==4.0.0
-psutil==5.9.0
-scikit-learn==1.1.2
-sparse==0.13.0
-tblib==1.7.0
-toolz==0.11.2
-typing-extensions==4.1.1
-uuid==1.30
-xgboost==1.4.2
-xxhash==3.0.0
\ No newline at end of file
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/MLmodel b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/MLmodel
similarity index 100%
rename from cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/MLmodel
rename to cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/MLmodel
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/conda.yaml b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/conda.yaml
similarity index 100%
rename from cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/conda.yaml
rename to cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/conda.yaml
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/model.pkl b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/model.pkl
similarity index 100%
rename from cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/model.pkl
rename to cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/model.pkl
diff --git a/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/requirements.txt b/cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/requirements.txt
similarity index 100%
rename from cli/endpoints/batch/deploy-models/heart-classifier-mlflow/deployment-custom/model/requirements.txt
rename to cli/endpoints/batch/deploy-models/heart-classifier-mlflow/model/requirements.txt
diff --git a/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver.py b/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver.py
index 5e8d611f34..a72a794d2d 100644
--- a/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver.py
+++ b/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver.py
@@ -19,7 +19,7 @@ def init():
     model_path = glob.glob(os.environ["AZUREML_MODEL_DIR"] + "/*/")[0]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(
diff --git a/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver_parquet.py b/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver_parquet.py
index 2cf82f970c..026aa86f8f 100644
--- a/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver_parquet.py
+++ b/sdk/python/endpoints/batch/deploy-models/heart-classifier-mlflow/code/batch_driver_parquet.py
@@ -22,7 +22,7 @@ def init():
     output_path = os.environ["AZUREML_BI_OUTPUT_PATH"]
 
     # Load the model, it's input types and output names
-    model = mlflow.pyfunc.load(model_path)
+    model = mlflow.pyfunc.load_model(model_path)
     if model.metadata and model.metadata.signature:
         if model.metadata.signature.inputs:
             model_input_types = dict(