Merge pull request #689 from GoogleCloudPlatform/update_vertex_xai

Add TF_USE_LEGACY_KERAS env variable to Vertex XAI notebook

Author: takumiohym

notebooks/responsible_ai/explainable_ai/labs/xai_image_vertex.ipynb

@@ -59,8 +59,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import base64\n",
     "import os\n",
+    "\n",
+    "os.environ[\"TF_USE_LEGACY_KERAS\"] = \"1\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import base64\n",
     "import random\n",
     "from datetime import datetime\n",
     "from io import BytesIO\n",
@@ -978,9 +988,9 @@
   },
   "environment": {
    "kernel": "conda-base-py",
-   "name": "workbench-notebooks.m121",
+   "name": "workbench-notebooks.m133",
    "type": "gcloud",
-   "uri": "us-docker.pkg.dev/deeplearning-platform-release/gcr.io/workbench-notebooks:m121"
+   "uri": "us-docker.pkg.dev/deeplearning-platform-release/gcr.io/workbench-notebooks:m133"
   },
   "kernelspec": {
    "display_name": "Python 3 (ipykernel) (Local)",
@@ -997,7 +1007,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.14"
+   "version": "3.10.18"
   }
  },
  "nbformat": 4,

notebooks/responsible_ai/explainable_ai/solutions/xai_image_vertex.ipynb

@@ -56,11 +56,25 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "tags": []
+   },
    "outputs": [],
    "source": [
-    "import base64\n",
     "import os\n",
+    "\n",
+    "os.environ[\"TF_USE_LEGACY_KERAS\"] = \"1\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "tags": []
+   },
+   "outputs": [],
+   "source": [
+    "import base64\n",
     "import random\n",
     "from datetime import datetime\n",
     "from io import BytesIO\n",