Fix model registry notebook

Author: aviruthen

v3-examples/ml-ops-examples/v3-model-registry-example/v3-model-registry-example.ipynb

@@ -185,18 +185,24 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from sagemaker.core.resources import ModelPackage\n",
+    "from sagemaker.serve.model_builder import ModelBuilder\n",
+    "\n",
+    "# Get the registered ModelPackage using the ARN as model_package_name\n",
+    "model_package = ModelPackage.get(model_package_name=registered_model_package_arn)\n",
+    "\n",
+    "s3_model_data_uri = model_package.inference_specification.containers[0].model_data_url\n",
+    "image_uri = model_package.inference_specification.containers[0].image\n",
     "\n",
-    "# Use the model_package in section above\n",
     "model_builder = ModelBuilder(\n",
-    "    model_package_arn=registered_model_package_arn,\n",
+    "    s3_model_data_url=s3_model_data_uri,\n",
+    "    image_uri=image_uri,\n",
     "    role_arn=role,\n",
     "    sagemaker_session=sagemaker_session\n",
     ")\n",
     "\n",
-    "\n",
-    "\n",
     "# Build the model\n",
-    "model = model_builder.build(model_name=\"my-model-from-registry\")"
+    "model = model_builder.build(model_name=\"my-model-from-registry\")\n"
    ]
   }
  ],