Addressing review feedback

Author: RyanMullins

site/en/responsible/docs/safeguards/shieldgemma2_on_huggingface.ipynb

@@ -105,6 +105,7 @@
       },
       "outputs": [],
       "source": [
+        "# @title install Hugging Face Transformers v4.50+\n",
         "! pip install -q 'transformers>=4.50.0'"
       ]
     },
@@ -116,7 +117,19 @@
       },
       "outputs": [],
       "source": [
-        "! huggingface-cli login"
+        "# @title Authenticate with Hugging Face Hub\n",
+        "# @markdown ShieldGemma is a gated model. To access the weights, you must accept\n",
+        "# @markdown the license on Hugging Face Hub under your account and then provide\n",
+        "# @markdown an [Access Token](https://huggingface.co/docs/hub/en/security-tokens)\n",
+        "# @markdown to authenticate with the Hugging Face Hub API. If using Colab, the\n",
+        "# @markdown easiest way to do this is by creating a read-only token specifically\n",
+        "# @markdown for Colab and setting this as the value of the `HF_TOKEN` secret;\n",
+        "# @markdown this token will then be reusable across all Colab notebooks. Other\n",
+        "# @markdown Python notebook platforms may provide a similar mechanism. For those\n",
+        "# @markdown that do not, un-comment the lines in this cell to install the\n",
+        "# @markdown Hugging Face Hub CLI and login in interactively.\n",
+        "# ! pip install -q 'huggingface_hub[cli]'\n",
+        "# ! huggingface-cli login"
       ]
     },
     {
@@ -188,7 +201,7 @@
         "\n",
         "# ShieldGemma prompts are constructed such that predicting the `Yes` token means\n",
         "# the content does violate the policy. If you are only interested in the\n",
-        "# violative condition, use to extract that slice from the output tensors.\n",
+        "# violative condition, you can extract only that slice from the output tensors.\n",
         "p_violated = scores.probabilities[:, 0]\n",
         "print(p_violated)\n"
       ]