Switch tensorflow models docs to use tf-models-official.

This works now that 2.9 (tensorflow and tf-models) is released.

PiperOrigin-RevId: 449782178

Author: MarkDaoust

site/en/tutorials/images/classification_with_model_garden.ipynb

@@ -1,5 +1,5 @@
 {
-  "cells": [ 
+  "cells": [
     {
       "cell_type": "markdown",
       "metadata": {
@@ -88,7 +88,9 @@
       "source": [
         "## Setup\n",
         "\n",
-        "Install and import the necessary modules. This tutorial uses the `tf-models-nightly` version of Model Garden."
+        "Install and import the necessary modules. This tutorial uses the `tf-models-nightly` version of Model Garden.\n",
+        "\n",
+        "Note: Upgrading TensorFlow to 2.9 in Colab breaks GPU support, so this colab is set to run on CPU until the Colab runtimes are updated."
       ]
     },
     {
@@ -100,7 +102,7 @@
       "outputs": [],
       "source": [
         "!pip uninstall -y opencv-python\n",
-        "!pip install -q tf-models-nightly"
+        "!pip install -U -q \"tensorflow>=2.9.0\" \"tf-models-official\""
       ]
     },
     {
@@ -149,8 +151,9 @@
       "source": [
         "import tensorflow_models as tfm\n",
         "\n",
-        "# Not in the tfm public API for v2.9. Will be available as `vision.serving` in v2.10\n",
-        "from official.vision.serving import export_saved_model_lib"
+        "# These are not in the tfm public API for v2.9. They will be available in v2.10\n",
+        "from official.vision.serving import export_saved_model_lib\n",
+        "import official.core.train_lib"
       ]
     },
     {
@@ -668,16 +671,14 @@
         "  show_batch(data['image'], data['label'], predictions)\n",
         "\n",
         "  if device=='CPU':\n",
-        "    plt.suptitle('The model was only trained for a few steps, it is not expected to do well.')"
+        "    plt.suptitle('The model was only trained for a few steps, it is not expected to do better than random.')"
       ]
     }
   ],
   "metadata": {
-    "accelerator": "GPU",
     "colab": {
       "collapsed_sections": [],
-      "name": "models_vision.ipynb",
-      "provenance": [],
+      "name": "classification_with_model_garden.ipynb",
       "toc_visible": true
     },
     "kernelspec": {