diff --git a/examples/notebooks/beam-ml/gemma_2_sentiment_and_summarization.ipynb b/examples/notebooks/beam-ml/gemma_2_sentiment_and_summarization.ipynb
index 64151d29992d..d7b2b157f613 100644
--- a/examples/notebooks/beam-ml/gemma_2_sentiment_and_summarization.ipynb
+++ b/examples/notebooks/beam-ml/gemma_2_sentiment_and_summarization.ipynb
@@ -161,14 +161,13 @@
       ]
     },
     {
-      "cell_type": "code",
-      "execution_count": null,
+      "cell_type": "markdown",
       "id": "29dOdG_kxzTv",
       "metadata": {
         "id": "29dOdG_kxzTv"
       },
-      "outputs": [],
       "source": [
+        "```sh\n",
         "ARG SERVING_BUILD_IMAGE=tensorflow/tensorflow:2.16.1-gpu\n",
         "\n",
         "FROM ${SERVING_BUILD_IMAGE}\n",
@@ -199,14 +198,13 @@
       ]
     },
     {
-      "cell_type": "code",
-      "execution_count": null,
+      "cell_type": "markdown",
       "id": "K4gJQ0e9pCR4",
       "metadata": {
         "id": "K4gJQ0e9pCR4"
       },
-      "outputs": [],
       "source": [
+        "```sh\n",
         "apache_beam[gcp]==2.54.0\n",
         "keras_nlp==0.14.3\n",
         "keras==3.4.1\n",
@@ -234,14 +232,13 @@
       ]
     },
     {
-      "cell_type": "code",
-      "execution_count": null,
+      "cell_type": "markdown",
       "id": "EdUxmUjqx58r",
       "metadata": {
         "id": "EdUxmUjqx58r"
       },
-      "outputs": [],
       "source": [
+        "```sh\n",
         "FROM <DOCKERFILE_IMAGE>\n",
         "\n",
         "WORKDIR /workspace\n",