diff --git a/examples/notebooks/beam-ml/automatic_model_refresh.ipynb b/examples/notebooks/beam-ml/automatic_model_refresh.ipynb
index 5b5d2ed484c0..7a5d660a99e9 100644
--- a/examples/notebooks/beam-ml/automatic_model_refresh.ipynb
+++ b/examples/notebooks/beam-ml/automatic_model_refresh.ipynb
@@ -130,9 +130,7 @@
 				]
 			}],
       "source": [
-        "!pip install apache_beam[gcp]>=2.46.0 --quiet\n",
-        "!pip install tensorflow==2.15.0 --quiet\n",
-        "!pip install tensorflow_hub --quiet"
+        "!pip install apache_beam[gcp]>=2.46.0  tensorflow==2.15.0 tensorflow_hub==0.16.1 keras==2.15.0 Pillow==11.0.0 --quiet"
       ]
     },
     {
@@ -272,7 +270,7 @@
       "cell_type": "code",
       "source": [
         "# In a requirements file, define the dependencies required for the pipeline.\n",
-        "!printf 'tensorflow==2.15.0\\ntensorflow_hub>=0.10.0\\nPillow>=9.0.0' > ./requirements.txt\n",
+        "!printf 'tensorflow==2.15.0\ntensorflow_hub==0.16.1\nkeras==2.15.0\nPillow==11.0.0' > ./requirements.txt\n",
         "# Install the pipeline dependencies on Dataflow.\n",
         "options.view_as(SetupOptions).requirements_file = './requirements.txt'"
       ],