Update responsibleaidashboard-text-classification-blbooksgenre.ipynb

Azure · Jul 12, 2024 · 771904b · 771904b
1 parent 631f7b7
commit 771904b
Showing 1 changed file with 7 additions and 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
@@ -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",