From af60671f8cd5b5e0de1b2b4a4623a3190cc14f75 Mon Sep 17 00:00:00 2001 From: Vertex MG Team Date: Wed, 26 Feb 2025 18:21:30 -0800 Subject: [PATCH] Update owlvit notebook to support Vertex Workbench. PiperOrigin-RevId: 731527931 --- .../model_garden_pytorch_owlvit.ipynb | 20 ++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb b/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb index 16b64e67c..a02c849f5 100644 --- a/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb +++ b/notebooks/community/model_garden/model_garden_pytorch_owlvit.ipynb @@ -34,6 +34,11 @@ "\n", "\n", " \n", + "
\n", + " \n", + " \"Workbench
Run in Workbench\n", + " \n", + " 		\n", " \n", " \"Google
Run in Colab Enterprise\n", " \n", @@ -122,6 +127,7 @@ "import matplotlib.pyplot as plt\n", "from google.cloud import aiplatform\n", "\n", + "! pip install --upgrade tensorflow\n", "! git clone https://github.com/GoogleCloudPlatform/vertex-ai-samples.git\n", "\n", "common_util = importlib.import_module(\n", @@ -134,6 +140,12 @@ "\n", "# Get the default region for launching jobs.\n", "if not REGION:\n", + " if \"GOOGLE_CLOUD_REGION\" not in os.environ:\n", + " raise ValueError(\n", + " \"REGION must be set. See\"\n", + " \" https://cloud.google.com/vertex-ai/docs/general/locations for\"\n", + " \" available cloud locations.\"\n", + " )\n", " REGION = os.environ[\"GOOGLE_CLOUD_REGION\"]\n", "\n", "# Enable the Vertex AI API and Compute Engine API, if not already.\n", @@ -227,7 +239,7 @@ " serving_env = {\n", " \"MODEL_ID\": model_id,\n", " \"TASK\": task,\n", - " \"DEPLOY_SOURCE\": \"notebook\",\n", + " \"DEPLOY_SOURCE\": common_util.get_deploy_source(),\n", " }\n", " # If the model_id is a GCS path, use artifact_uri to pass it to serving docker.\n", " artifact_uri = model_id if model_id.startswith(\"gs://\") else None\n", @@ -239,7 +251,7 @@ " serving_container_health_route=\"/ping\",\n", " serving_container_environment_variables=serving_env,\n", " artifact_uri=artifact_uri,\n", - " model_garden_source_model_name=\"publishers/google/models/owlvit-base-patch32\",\n", + " model_garden_source_model_name=(\"publishers/google/models/owlvit-base-patch32\"),\n", " )\n", " model.deploy(\n", " endpoint=endpoint,\n", @@ -295,7 +307,9 @@ "# @markdown Once deployment succeeds, you can generate Object Detection based\n", "# @markdown on the input image and the text.\n", "\n", - "image_url = \"http://images.cocodataset.org/val2017/000000039769.jpg\" # @param {type:\"string\"}\n", + "image_url = (\n", + " \"http://images.cocodataset.org/val2017/000000039769.jpg\" # @param {type:\"string\"}\n", + ")\n", "text = \"cat\" # @param {type:\"string\"}\n", "image = common_util.download_image(image_url)\n", "\n",