fix: missing variable

Author: meilame-tayebjee

notebooks/example.ipynb

@@ -430,9 +430,10 @@
    "outputs": [],
    "source": [
     "# test the TextEmbedder: it takes as input a tensor of token ids and outputs a tensor of embeddings\n",
+    "\n",
     "text_embedder_output = text_embedder(input_ids=batch[\"input_ids\"], attention_mask=batch[\"attention_mask\"])\n",
     "\n",
-    "print(\"TextEmbedder input: \", text_embedder_input.input_ids)\n",
+    "print(\"TextEmbedder input: \", batch[\"input_ids\"])\n",
     "print(\"TextEmbedder output shape: \", text_embedder_output.shape)"
    ]
   },