From 771904bfad4093f23f04a6ca05fed83f29e26d5f Mon Sep 17 00:00:00 2001
From: tongy-msft <91754176+tongyu-microsoft@users.noreply.github.com>
Date: Fri, 12 Jul 2024 11:40:58 -0700
Subject: [PATCH] Update
 responsibleaidashboard-text-classification-blbooksgenre.ipynb

---
 ...bleaidashboard-text-classification-blbooksgenre.ipynb | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb b/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb
index 61f3b4839d..a91871b725 100644
--- a/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb
+++ b/sdk/python/responsible-ai/text/responsibleaidashboard-text-classification-blbooksgenre.ipynb
@@ -128,7 +128,11 @@
    "source": [
     "def load_dataset(split):\n",
     "    config_kwargs = {\"name\": \"title_genre_classifiction\"}\n",
-    "    dataset = datasets.load_dataset(\"blbooksgenre\", split=split, trust_remote_code=True, **config_kwargs)\n",
+    "    dataset = datasets.load_dataset(\n",
+    "        \"blbooksgenre\",\n",
+    "        split=split,\n",
+    "        trust_remote_code=True,\n",
+    "        **config_kwargs)\n",
     "    return pd.DataFrame({\"text\": dataset[\"title\"], \"label\": dataset[\"label\"]})\n",
     "\n",
     "\n",
@@ -447,7 +451,8 @@
     "                   max_retries=max_retries,\n",
     "                   retry_delay=retry_delay)\n",
     "    model = AutoModelForSequenceClassification.from_pretrained(\n",
-    "        BLBOOKSGENRE_MODEL_NAME, num_labels=NUM_LABELS)\n",
+    "        BLBOOKSGENRE_MODEL_NAME, num_labels=NUM_LABELS,\n",
+    "        trust_remote_code=True)\n",
     "    return model\n",
     "\n",
     "\n",