minor fix

Author: Michael Tsang

demos/1. text analysis/demo_bert_torch_word_level.ipynb

@@ -122,9 +122,9 @@
     "            flattened_token_ids.extend(id_group)\n",
     "        return flattened_token_ids\n",
     "    \n",
-    "def get_token_id_groups_from_words_in_input(text, tokenizer):\n",
+    "def get_token_id_groups_from_words_in_input(text, baseline_token, tokenizer):\n",
     "    # get default start and end ids of a sentence\n",
-    "    text_ids, _ = get_input_baseline_ids(text, \"\", tokenizer)\n",
+    "    text_ids, baseline_ids = get_input_baseline_ids(text, baseline_token, tokenizer)\n",
     "    start_id, end_id = text_ids[0], text_ids[-1]\n",
     "\n",
     "    # get each word in the input text as a list of corresponding token ids\n",
@@ -145,7 +145,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "token_id_groups, baseline_id_groups = get_token_id_groups_from_words_in_input(text, tokenizer)\n",
+    "token_id_groups, baseline_id_groups = get_token_id_groups_from_words_in_input(text, baseline_token, tokenizer)\n",
     "\n",
     "xf = WordXformer(token_id_groups, baseline_id_groups) \n",
     "apgo = Archipelago(model_wrapper, data_xformer=xf, output_indices=class_idx, batch_size=20)"