diff --git a/notebooks/astradb_haystack_integration.ipynb b/notebooks/astradb_haystack_integration.ipynb
index 4d78f64..1cf71a8 100644
--- a/notebooks/astradb_haystack_integration.ipynb
+++ b/notebooks/astradb_haystack_integration.ipynb
@@ -1,1028 +1,1028 @@
 {
-  "cells": [
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "qFBDARhK0dn5"
-      },
-      "source": [
-        "## Introduction\n",
-        "\n",
-        "In this notebook, you'll learn how to use [AstraDB](https://docs.datastax.com/en/astra-serverless/docs/) as a data source in your Haystack pipelines."
-      ]
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "qFBDARhK0dn5"
+   },
+   "source": [
+    "## Introduction\n",
+    "\n",
+    "In this notebook, you'll learn how to use [AstraDB](https://docs.datastax.com/en/astra-serverless/docs/) as a data source in your Haystack pipelines."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "JT4_BeDjZcCI"
+   },
+   "source": [
+    "# Prerequisites\n",
+    "\n",
+    "You'll need an [OpenAPI key](https://help.openai.com/en/articles/4936850-where-do-i-find-my-api-key) to follow along. (Haystack is model-agnostic so feel free to use a different one if you'd prefer!)\n",
+    "\n",
+    "You'll need the following variables in order to use the Haystack extension. The following tutorials will show you how to create an AstraDB database, and save these pieces of information.\n",
+    "\n",
+    "- API Endpoint\n",
+    "- Token\n",
+    "- Astra keyspace\n",
+    "- Astra collection name\n",
+    "\n",
+    "Follow the first step in this [this tutorial to create a free AstraDB database](https://docs.datastax.com/en/astra-serverless/docs/manage/db/manage-create.html) and save your database ID, application token, keyspace, and database region.\n",
+    "\n",
+    "[Follow these steps to create a collection](https://docs.datastax.com/en/astra/astra-db-vector/databases/manage-collections.html). Save the name of your collection.\n",
+    "\n",
+    "Choose the number of dimensions that matches the [embedding model](https://haystack.deepset.ai/blog/what-is-text-vectorization-in-nlp) you plan on using. For this example we'll use a 384-dimension model, [`sentence-transformers/all-MiniLM-L6-v2`](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2)."
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "9qpgbYEB787y"
+   },
+   "source": [
+    "Next, install our dependencies."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "colab": {
+     "base_uri": "https://localhost:8080/"
     },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "JT4_BeDjZcCI"
-      },
-      "source": [
-        "# Prerequisites\n",
-        "\n",
-        "You'll need an [OpenAPI key](https://help.openai.com/en/articles/4936850-where-do-i-find-my-api-key) to follow along. (Haystack is model-agnostic so feel free to use a different one if you'd prefer!)\n",
-        "\n",
-        "You'll need the following variables in order to use the Haystack extension. The following tutorials will show you how to create an AstraDB database, and save these pieces of information.\n",
-        "\n",
-        "- API Endpoint\n",
-        "- Token\n",
-        "- Astra keyspace\n",
-        "- Astra collection name\n",
-        "\n",
-        "Follow the first step in this [this tutorial to create a free AstraDB database](https://docs.datastax.com/en/astra-serverless/docs/manage/db/manage-create.html) and save your database ID, application token, keyspace, and database region.\n",
-        "\n",
-        "[Follow these steps to create a collection](https://docs.datastax.com/en/astra/astra-db-vector/databases/manage-collections.html). Save the name of your collection.\n",
-        "\n",
-        "Choose the number of dimensions that matches the [embedding model](https://haystack.deepset.ai/blog/what-is-text-vectorization-in-nlp) you plan on using. For this example we'll use a 384-dimension model, [`sentence-transformers/all-MiniLM-L6-v2`](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2)."
-      ]
+    "id": "BtIZl9jGm6QP",
+    "outputId": "2940fb89-b935-4dd1-b66f-c7a106cc06ec"
+   },
+   "outputs": [],
+   "source": [
+    "!pip install astra-haystack sentence-transformers"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "ipzOS2-O5G9b"
+   },
+   "source": [
+    "Here you'll enter your credentials and such. In production code, you'd want to use environment variables for sensitive credentials such as the application token to avoid committing those to source control.\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "colab": {
+     "base_uri": "https://localhost:8080/"
     },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "9qpgbYEB787y"
-      },
-      "source": [
-        "Next, install our dependencies."
-      ]
+    "id": "_EKlRTHucc2j",
+    "outputId": "402df0d2-a4cd-4e05-f3ab-51c376147e74"
+   },
+   "outputs": [],
+   "source": [
+    "from getpass import getpass\n",
+    "import os\n",
+    "\n",
+    "os.environ[\"OPENAI_API_KEY\"] = getpass(\"Enter your openAI key:\")\n",
+    "os.environ[\"ASTRA_DB_API_ENDPOINT\"] = getpass(\"Enter your Astra API Endpoint:\")\n",
+    "os.environ[\"ASTRA_DB_APPLICATION_TOKEN\"] = getpass(\"Enter your Astra application token (e.g.AstraCS:xxx ):\")\n",
+    "ASTRA_DB_COLLECTION_NAME = getpass(\"enter your Astra collection name:\")\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "d3WEbADaAfzI"
+   },
+   "source": [
+    "Next we'll create a Haystack pipeline to create the embeddings and add them into the `AstraDocumentStore`."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "colab": {
+     "base_uri": "https://localhost:8080/",
+     "height": 85,
+     "referenced_widgets": [
+      "d441033160e4421bb9af18d941d50d6b",
+      "fb58298ecdf543389e615ccdf7e6a37a",
+      "cc332ffcf5e5401ab26efef1b35fc34c",
+      "3aa85f749d99477da52d8c3693046e6a",
+      "ce61f00f6ff944d695002f7a8b599294",
+      "a9d1bb3132a54b52a3f2f16aec5b94a7",
+      "8208352647db4305bec1c2ab9499471b",
+      "1518245edc554442bd0ad6804273309c",
+      "dadf5298941e41ad93e19aea51e3008f",
+      "b4199cf8555a44cb80d22645750d0d32",
+      "9971ba728ebf44b2a2644c3930981ef4"
+     ]
     },
+    "id": "A1iA_0dmXV3L",
+    "outputId": "0506858c-83f8-4bb5-c77f-bfc0b22a6589"
+   },
+   "outputs": [
     {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "id": "BtIZl9jGm6QP",
-        "outputId": "2940fb89-b935-4dd1-b66f-c7a106cc06ec"
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "d441033160e4421bb9af18d941d50d6b",
+       "version_major": 2,
+       "version_minor": 0
       },
-      "outputs": [],
-      "source": [
-        "!pip install astra-haystack sentence-transformers"
+      "text/plain": [
+       "Batches:   0%|          | 0/1 [00:00<?, ?it/s]"
       ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "ipzOS2-O5G9b"
-      },
-      "source": [
-        "Here you'll enter your credentials and such. In production code, you'd want to use environment variables for sensitive credentials such as the application token to avoid committing those to source control.\n"
-      ]
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "WARNING:astra_haystack.document_store:No documents written. Argument policy set to SKIP\n"
+     ]
     },
     {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "id": "_EKlRTHucc2j",
-        "outputId": "402df0d2-a4cd-4e05-f3ab-51c376147e74"
-      },
-      "outputs": [],
-      "source": [
-        "from getpass import getpass\n",
-        "import os\n",
-        "\n",
-        "os.environ[\"OPENAI_API_KEY\"] = getpass(\"Enter your openAI key:\")\n",
-        "os.environ[\"ASTRA_DB_API_ENDPOINT\"] = getpass(\"Enter your Astra API Endpoint:\")\n",
-        "os.environ[\"ASTRA_DB_APPLICATION_TOKEN\"] = getpass(\"Enter your Astra application token (e.g.AstraCS:xxx ):\")\n",
-        "ASTRA_DB_COLLECTION_NAME = getpass(\"enter your Astra collection name:\")\n"
-      ]
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "3\n"
+     ]
+    }
+   ],
+   "source": [
+    "import logging\n",
+    "\n",
+    "from haystack import Document, Pipeline\n",
+    "\n",
+    "from haystack.components.embedders import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder\n",
+    "from haystack.components.writers import DocumentWriter\n",
+    "from haystack.document_stores.types import DuplicatePolicy\n",
+    "\n",
+    "from haystack_integrations.document_stores.astra import AstraDocumentStore\n",
+    "\n",
+    "logger = logging.getLogger(__name__)\n",
+    "logging.basicConfig(level=logging.INFO)\n",
+    "\n",
+    "embedding_model_name = \"sentence-transformers/all-MiniLM-L6-v2\"\n",
+    "\n",
+    "# embedding_dim is the number of dimensions the embedding model supports.\n",
+    "document_store = AstraDocumentStore(\n",
+    "    collection_name=ASTRA_DB_COLLECTION_NAME,\n",
+    "    duplicates_policy=DuplicatePolicy.SKIP,\n",
+    "    embedding_dimension=384,\n",
+    ")\n",
+    "\n",
+    "\n",
+    "# Add Documents\n",
+    "documents = [\n",
+    "    Document(content=\"There are over 7,000 languages spoken around the world today.\"),\n",
+    "    Document(\n",
+    "        content=\"Elephants have been observed to behave in a way that indicates\"\n",
+    "        \" a high level of self-awareness, such as recognizing themselves in mirrors.\"\n",
+    "    ),\n",
+    "    Document(\n",
+    "        content=\"In certain parts of the world, like the Maldives, Puerto Rico, \"\n",
+    "        \"and San Diego, you can witness the phenomenon of bioluminescent waves.\"\n",
+    "    ),\n",
+    "]\n",
+    "index_pipeline = Pipeline()\n",
+    "index_pipeline.add_component(\n",
+    "    instance=SentenceTransformersDocumentEmbedder(model=embedding_model_name),\n",
+    "    name=\"embedder\",\n",
+    ")\n",
+    "index_pipeline.add_component(instance=DocumentWriter(document_store=document_store, policy=DuplicatePolicy.SKIP), name=\"writer\")\n",
+    "index_pipeline.connect(\"embedder.documents\", \"writer.documents\")\n",
+    "\n",
+    "index_pipeline.run({\"embedder\": {\"documents\": documents}})\n",
+    "\n",
+    "print(document_store.count_documents())\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "d-xsXsWwFwo1"
+   },
+   "source": [
+    "Next we'll make a RAG pipeline so we can query our documents."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "colab": {
+     "base_uri": "https://localhost:8080/",
+     "height": 87,
+     "referenced_widgets": [
+      "836eef274ba14b01b4cad3a226a33d6f",
+      "1976fec3079d4bac8c175c53f3f4ec32",
+      "b4fb8c6b9b424fa79560233caa9d8b40",
+      "6f8fba6461ac4109b7bf6316967f316d",
+      "e110936ae62d471d8bb90ff1824ed9ad",
+      "b3b3c648b29e4019b83db9b1483416c2",
+      "c93290da6a994795ae1e01e082cb2c66",
+      "4ce46f0cb05641759d25207ced6cfe94",
+      "03d9e42d6ecc4a269c5be231a0955f0f",
+      "c8d28cdf27fb43ddb41c83357775afcf",
+      "c7183e1571594b049c4910a30c89be7f"
+     ]
     },
+    "id": "CgSk0bxPFUxx",
+    "outputId": "4910fcc2-da85-4e5a-a1bb-1a39661846e1"
+   },
+   "outputs": [
     {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "d3WEbADaAfzI"
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "836eef274ba14b01b4cad3a226a33d6f",
+       "version_major": 2,
+       "version_minor": 0
       },
-      "source": [
-        "Next we'll create a Haystack pipeline to create the embeddings and add them into the `AstraDocumentStore`."
+      "text/plain": [
+       "Batches:   0%|          | 0/1 [00:00<?, ?it/s]"
       ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
     },
     {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "colab": {
-          "base_uri": "https://localhost:8080/",
-          "height": 85,
-          "referenced_widgets": [
-            "d441033160e4421bb9af18d941d50d6b",
-            "fb58298ecdf543389e615ccdf7e6a37a",
-            "cc332ffcf5e5401ab26efef1b35fc34c",
-            "3aa85f749d99477da52d8c3693046e6a",
-            "ce61f00f6ff944d695002f7a8b599294",
-            "a9d1bb3132a54b52a3f2f16aec5b94a7",
-            "8208352647db4305bec1c2ab9499471b",
-            "1518245edc554442bd0ad6804273309c",
-            "dadf5298941e41ad93e19aea51e3008f",
-            "b4199cf8555a44cb80d22645750d0d32",
-            "9971ba728ebf44b2a2644c3930981ef4"
-          ]
-        },
-        "id": "A1iA_0dmXV3L",
-        "outputId": "0506858c-83f8-4bb5-c77f-bfc0b22a6589"
-      },
-      "outputs": [
-        {
-          "data": {
-            "application/vnd.jupyter.widget-view+json": {
-              "model_id": "d441033160e4421bb9af18d941d50d6b",
-              "version_major": 2,
-              "version_minor": 0
-            },
-            "text/plain": [
-              "Batches:   0%|          | 0/1 [00:00<?, ?it/s]"
-            ]
-          },
-          "metadata": {},
-          "output_type": "display_data"
-        },
-        {
-          "name": "stderr",
-          "output_type": "stream",
-          "text": [
-            "WARNING:astra_haystack.document_store:No documents written. Argument policy set to SKIP\n"
-          ]
-        },
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "3\n"
-          ]
-        }
-      ],
-      "source": [
-        "import logging\n",
-        "\n",
-        "from haystack import Document, Pipeline\n",
-        "\n",
-        "from haystack.components.embedders import SentenceTransformersDocumentEmbedder, SentenceTransformersTextEmbedder\n",
-        "from haystack.components.writers import DocumentWriter\n",
-        "from haystack.document_stores.types import DuplicatePolicy\n",
-        "\n",
-        "from haystack_integrations.document_stores.astra import AstraDocumentStore\n",
-        "\n",
-        "logger = logging.getLogger(__name__)\n",
-        "logging.basicConfig(level=logging.INFO)\n",
-        "\n",
-        "embedding_model_name = \"sentence-transformers/all-MiniLM-L6-v2\"\n",
-        "\n",
-        "# embedding_dim is the number of dimensions the embedding model supports.\n",
-        "document_store = AstraDocumentStore(\n",
-        "    astra_collection=ASTRA_DB_COLLECTION_NAME,\n",
-        "    duplicates_policy=DuplicatePolicy.SKIP,\n",
-        "    embedding_dim=384,\n",
-        ")\n",
-        "\n",
-        "\n",
-        "# Add Documents\n",
-        "documents = [\n",
-        "    Document(content=\"There are over 7,000 languages spoken around the world today.\"),\n",
-        "    Document(\n",
-        "        content=\"Elephants have been observed to behave in a way that indicates\"\n",
-        "        \" a high level of self-awareness, such as recognizing themselves in mirrors.\"\n",
-        "    ),\n",
-        "    Document(\n",
-        "        content=\"In certain parts of the world, like the Maldives, Puerto Rico, \"\n",
-        "        \"and San Diego, you can witness the phenomenon of bioluminescent waves.\"\n",
-        "    ),\n",
-        "]\n",
-        "index_pipeline = Pipeline()\n",
-        "index_pipeline.add_component(\n",
-        "    instance=SentenceTransformersDocumentEmbedder(model=embedding_model_name),\n",
-        "    name=\"embedder\",\n",
-        ")\n",
-        "index_pipeline.add_component(instance=DocumentWriter(document_store=document_store, policy=DuplicatePolicy.SKIP), name=\"writer\")\n",
-        "index_pipeline.connect(\"embedder.documents\", \"writer.documents\")\n",
-        "\n",
-        "index_pipeline.run({\"embedder\": {\"documents\": documents}})\n",
-        "\n",
-        "print(document_store.count_documents())\n"
-      ]
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "{'answer_builder': {'answers': [GeneratedAnswer(data='There are over 7,000 languages spoken around the world today.', query='How many languages are there in the world today?', documents=[Document(id=cfe93bc1c274908801e6670440bf2bbba54fad792770d57421f85ffa2a4fcc94, content: 'There are over 7,000 languages spoken around the world today.', score: 0.9267925, embedding: vector of size 384), Document(id=6f20658aeac3c102495b198401c1c0c2bd71d77b915820304d4fbc324b2f3cdb, content: 'Elephants have been observed to behave in a way that indicates a high level of self-awareness, such ...', score: 0.5357444, embedding: vector of size 384)], meta={'model': 'gpt-4o-mini-2024-07-18', 'index': 0, 'finish_reason': 'stop', 'usage': {'completion_tokens': 14, 'prompt_tokens': 83, 'total_tokens': 97}})]}}\n"
+     ]
+    }
+   ],
+   "source": [
+    "from haystack.components.builders.answer_builder import AnswerBuilder\n",
+    "from haystack.components.builders.prompt_builder import PromptBuilder\n",
+    "from haystack.components.generators import OpenAIGenerator\n",
+    "from haystack_integrations.components.retrievers.astra import AstraEmbeddingRetriever\n",
+    "\n",
+    "prompt_template = \"\"\"\n",
+    "                Given these documents, answer the question.\n",
+    "                Documents:\n",
+    "                {% for doc in documents %}\n",
+    "                    {{ doc.content }}\n",
+    "                {% endfor %}\n",
+    "                Question: {{question}}\n",
+    "                Answer:\n",
+    "                \"\"\"\n",
+    "\n",
+    "rag_pipeline = Pipeline()\n",
+    "rag_pipeline.add_component(\n",
+    "    instance=SentenceTransformersTextEmbedder(model=embedding_model_name),\n",
+    "    name=\"embedder\",\n",
+    ")\n",
+    "rag_pipeline.add_component(instance=AstraEmbeddingRetriever(document_store=document_store), name=\"retriever\")\n",
+    "rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name=\"prompt_builder\")\n",
+    "rag_pipeline.add_component(instance=OpenAIGenerator(), name=\"llm\")\n",
+    "rag_pipeline.add_component(instance=AnswerBuilder(), name=\"answer_builder\")\n",
+    "rag_pipeline.connect(\"embedder\", \"retriever\")\n",
+    "rag_pipeline.connect(\"retriever\", \"prompt_builder.documents\")\n",
+    "rag_pipeline.connect(\"prompt_builder\", \"llm\")\n",
+    "rag_pipeline.connect(\"llm.replies\", \"answer_builder.replies\")\n",
+    "rag_pipeline.connect(\"llm.meta\", \"answer_builder.meta\")\n",
+    "rag_pipeline.connect(\"retriever\", \"answer_builder.documents\")\n",
+    "\n",
+    "\n",
+    "# Draw the pipeline\n",
+    "rag_pipeline.draw(path=\"./rag_pipeline.png\")\n",
+    "\n",
+    "\n",
+    "# Run the pipeline\n",
+    "question = \"How many languages are there in the world today?\"\n",
+    "result = rag_pipeline.run(\n",
+    "    {\n",
+    "        \"embedder\": {\"text\": question},\n",
+    "        \"retriever\": {\"top_k\": 2},\n",
+    "        \"prompt_builder\": {\"question\": question},\n",
+    "        \"answer_builder\": {\"query\": question},\n",
+    "    }\n",
+    ")\n",
+    "\n",
+    "print(result)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {
+    "id": "r05kV1K-G1Ar"
+   },
+   "source": [
+    "The output should be something like this:\n",
+    "```bash\n",
+    "{'answer_builder': {'answers': [GeneratedAnswer(data='There are over 7,000 languages spoken around the world today.', query='How many languages are there in the world today?', documents=[Document(id=cfe93bc1c274908801e6670440bf2bbba54fad792770d57421f85ffa2a4fcc94, content: 'There are over 7,000 languages spoken around the world today.', score: 0.9267925, embedding: vector of size 384), Document(id=6f20658aeac3c102495b198401c1c0c2bd71d77b915820304d4fbc324b2f3cdb, content: 'Elephants have been observed to behave in a way that indicates a high level of self-awareness, such ...', score: 0.5357444, embedding: vector of size 384)], meta={'model': 'gpt-4o-mini-2024-07-18', 'index': 0, 'finish_reason': 'stop', 'usage': {'completion_tokens': 14, 'prompt_tokens': 83, 'total_tokens': 97}})]}}\n",
+    "```\n",
+    "\n",
+    "Now that you understand how to use AstraDB as a data source for your Haystack pipeline. Thanks for reading! To learn more about Haystack, [join us on Discord](https://discord.gg/QMP5jgMH) or [sign up for our Monthly newsletter](https://landing.deepset.ai/haystack-community-updates?utm_campaign=developer-relations&utm_source=astradb-haystack-notebook)."
+   ]
+  }
+ ],
+ "metadata": {
+  "colab": {
+   "provenance": []
+  },
+  "kernelspec": {
+   "display_name": "Python 3",
+   "name": "python3"
+  },
+  "language_info": {
+   "name": "python"
+  },
+  "widgets": {
+   "application/vnd.jupyter.widget-state+json": {
+    "03d9e42d6ecc4a269c5be231a0955f0f": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "ProgressStyleModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "ProgressStyleModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "StyleView",
+      "bar_color": null,
+      "description_width": ""
+     }
     },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "d-xsXsWwFwo1"
-      },
-      "source": [
-        "Next we'll make a RAG pipeline so we can query our documents."
-      ]
+    "1518245edc554442bd0ad6804273309c": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
     },
-    {
-      "cell_type": "code",
-      "execution_count": null,
-      "metadata": {
-        "colab": {
-          "base_uri": "https://localhost:8080/",
-          "height": 87,
-          "referenced_widgets": [
-            "836eef274ba14b01b4cad3a226a33d6f",
-            "1976fec3079d4bac8c175c53f3f4ec32",
-            "b4fb8c6b9b424fa79560233caa9d8b40",
-            "6f8fba6461ac4109b7bf6316967f316d",
-            "e110936ae62d471d8bb90ff1824ed9ad",
-            "b3b3c648b29e4019b83db9b1483416c2",
-            "c93290da6a994795ae1e01e082cb2c66",
-            "4ce46f0cb05641759d25207ced6cfe94",
-            "03d9e42d6ecc4a269c5be231a0955f0f",
-            "c8d28cdf27fb43ddb41c83357775afcf",
-            "c7183e1571594b049c4910a30c89be7f"
-          ]
-        },
-        "id": "CgSk0bxPFUxx",
-        "outputId": "4910fcc2-da85-4e5a-a1bb-1a39661846e1"
-      },
-      "outputs": [
-        {
-          "data": {
-            "application/vnd.jupyter.widget-view+json": {
-              "model_id": "836eef274ba14b01b4cad3a226a33d6f",
-              "version_major": 2,
-              "version_minor": 0
-            },
-            "text/plain": [
-              "Batches:   0%|          | 0/1 [00:00<?, ?it/s]"
-            ]
-          },
-          "metadata": {},
-          "output_type": "display_data"
-        },
-        {
-          "name": "stdout",
-          "output_type": "stream",
-          "text": [
-            "{'answer_builder': {'answers': [GeneratedAnswer(data='There are over 7,000 languages spoken around the world today.', query='How many languages are there in the world today?', documents=[Document(id=cfe93bc1c274908801e6670440bf2bbba54fad792770d57421f85ffa2a4fcc94, content: 'There are over 7,000 languages spoken around the world today.', score: 0.9267925, embedding: vector of size 384), Document(id=6f20658aeac3c102495b198401c1c0c2bd71d77b915820304d4fbc324b2f3cdb, content: 'Elephants have been observed to behave in a way that indicates a high level of self-awareness, such ...', score: 0.5357444, embedding: vector of size 384)], meta={'model': 'gpt-4o-mini-2024-07-18', 'index': 0, 'finish_reason': 'stop', 'usage': {'completion_tokens': 14, 'prompt_tokens': 83, 'total_tokens': 97}})]}}\n"
-          ]
-        }
+    "1976fec3079d4bac8c175c53f3f4ec32": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "HTMLModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "HTMLModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "HTMLView",
+      "description": "",
+      "description_tooltip": null,
+      "layout": "IPY_MODEL_b3b3c648b29e4019b83db9b1483416c2",
+      "placeholder": "​",
+      "style": "IPY_MODEL_c93290da6a994795ae1e01e082cb2c66",
+      "value": "Batches: 100%"
+     }
+    },
+    "3aa85f749d99477da52d8c3693046e6a": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "HTMLModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "HTMLModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "HTMLView",
+      "description": "",
+      "description_tooltip": null,
+      "layout": "IPY_MODEL_b4199cf8555a44cb80d22645750d0d32",
+      "placeholder": "​",
+      "style": "IPY_MODEL_9971ba728ebf44b2a2644c3930981ef4",
+      "value": " 1/1 [00:00&lt;00:00,  7.31it/s]"
+     }
+    },
+    "4ce46f0cb05641759d25207ced6cfe94": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
+    },
+    "6f8fba6461ac4109b7bf6316967f316d": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "HTMLModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "HTMLModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "HTMLView",
+      "description": "",
+      "description_tooltip": null,
+      "layout": "IPY_MODEL_c8d28cdf27fb43ddb41c83357775afcf",
+      "placeholder": "​",
+      "style": "IPY_MODEL_c7183e1571594b049c4910a30c89be7f",
+      "value": " 1/1 [00:00&lt;00:00, 16.92it/s]"
+     }
+    },
+    "8208352647db4305bec1c2ab9499471b": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "DescriptionStyleModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "DescriptionStyleModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "StyleView",
+      "description_width": ""
+     }
+    },
+    "836eef274ba14b01b4cad3a226a33d6f": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "HBoxModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "HBoxModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "HBoxView",
+      "box_style": "",
+      "children": [
+       "IPY_MODEL_1976fec3079d4bac8c175c53f3f4ec32",
+       "IPY_MODEL_b4fb8c6b9b424fa79560233caa9d8b40",
+       "IPY_MODEL_6f8fba6461ac4109b7bf6316967f316d"
       ],
-      "source": [
-        "from haystack.components.builders.answer_builder import AnswerBuilder\n",
-        "from haystack.components.builders.prompt_builder import PromptBuilder\n",
-        "from haystack.components.generators import OpenAIGenerator\n",
-        "from haystack_integrations.components.retrievers.astra import AstraEmbeddingRetriever\n",
-        "\n",
-        "prompt_template = \"\"\"\n",
-        "                Given these documents, answer the question.\n",
-        "                Documents:\n",
-        "                {% for doc in documents %}\n",
-        "                    {{ doc.content }}\n",
-        "                {% endfor %}\n",
-        "                Question: {{question}}\n",
-        "                Answer:\n",
-        "                \"\"\"\n",
-        "\n",
-        "rag_pipeline = Pipeline()\n",
-        "rag_pipeline.add_component(\n",
-        "    instance=SentenceTransformersTextEmbedder(model=embedding_model_name),\n",
-        "    name=\"embedder\",\n",
-        ")\n",
-        "rag_pipeline.add_component(instance=AstraEmbeddingRetriever(document_store=document_store), name=\"retriever\")\n",
-        "rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name=\"prompt_builder\")\n",
-        "rag_pipeline.add_component(instance=OpenAIGenerator(), name=\"llm\")\n",
-        "rag_pipeline.add_component(instance=AnswerBuilder(), name=\"answer_builder\")\n",
-        "rag_pipeline.connect(\"embedder\", \"retriever\")\n",
-        "rag_pipeline.connect(\"retriever\", \"prompt_builder.documents\")\n",
-        "rag_pipeline.connect(\"prompt_builder\", \"llm\")\n",
-        "rag_pipeline.connect(\"llm.replies\", \"answer_builder.replies\")\n",
-        "rag_pipeline.connect(\"llm.meta\", \"answer_builder.meta\")\n",
-        "rag_pipeline.connect(\"retriever\", \"answer_builder.documents\")\n",
-        "\n",
-        "\n",
-        "# Draw the pipeline\n",
-        "rag_pipeline.draw(\"./rag_pipeline.png\")\n",
-        "\n",
-        "\n",
-        "# Run the pipeline\n",
-        "question = \"How many languages are there in the world today?\"\n",
-        "result = rag_pipeline.run(\n",
-        "    {\n",
-        "        \"embedder\": {\"text\": question},\n",
-        "        \"retriever\": {\"top_k\": 2},\n",
-        "        \"prompt_builder\": {\"question\": question},\n",
-        "        \"answer_builder\": {\"query\": question},\n",
-        "    }\n",
-        ")\n",
-        "\n",
-        "print(result)"
-      ]
+      "layout": "IPY_MODEL_e110936ae62d471d8bb90ff1824ed9ad"
+     }
     },
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "r05kV1K-G1Ar"
-      },
-      "source": [
-        "The output should be something like this:\n",
-        "```bash\n",
-        "{'answer_builder': {'answers': [GeneratedAnswer(data='There are over 7,000 languages spoken around the world today.', query='How many languages are there in the world today?', documents=[Document(id=cfe93bc1c274908801e6670440bf2bbba54fad792770d57421f85ffa2a4fcc94, content: 'There are over 7,000 languages spoken around the world today.', score: 0.9267925, embedding: vector of size 384), Document(id=6f20658aeac3c102495b198401c1c0c2bd71d77b915820304d4fbc324b2f3cdb, content: 'Elephants have been observed to behave in a way that indicates a high level of self-awareness, such ...', score: 0.5357444, embedding: vector of size 384)], meta={'model': 'gpt-4o-mini-2024-07-18', 'index': 0, 'finish_reason': 'stop', 'usage': {'completion_tokens': 14, 'prompt_tokens': 83, 'total_tokens': 97}})]}}\n",
-        "```\n",
-        "\n",
-        "Now that you understand how to use AstraDB as a data source for your Haystack pipeline. Thanks for reading! To learn more about Haystack, [join us on Discord](https://discord.gg/QMP5jgMH) or [sign up for our Monthly newsletter](https://landing.deepset.ai/haystack-community-updates?utm_campaign=developer-relations&utm_source=astradb-haystack-notebook)."
-      ]
-    }
-  ],
-  "metadata": {
-    "colab": {
-      "provenance": []
+    "9971ba728ebf44b2a2644c3930981ef4": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "DescriptionStyleModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "DescriptionStyleModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "StyleView",
+      "description_width": ""
+     }
+    },
+    "a9d1bb3132a54b52a3f2f16aec5b94a7": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
+    },
+    "b3b3c648b29e4019b83db9b1483416c2": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
+    },
+    "b4199cf8555a44cb80d22645750d0d32": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
+    },
+    "b4fb8c6b9b424fa79560233caa9d8b40": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "FloatProgressModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "FloatProgressModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "ProgressView",
+      "bar_style": "success",
+      "description": "",
+      "description_tooltip": null,
+      "layout": "IPY_MODEL_4ce46f0cb05641759d25207ced6cfe94",
+      "max": 1,
+      "min": 0,
+      "orientation": "horizontal",
+      "style": "IPY_MODEL_03d9e42d6ecc4a269c5be231a0955f0f",
+      "value": 1
+     }
+    },
+    "c7183e1571594b049c4910a30c89be7f": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "DescriptionStyleModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "DescriptionStyleModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "StyleView",
+      "description_width": ""
+     }
     },
-    "kernelspec": {
-      "display_name": "Python 3",
-      "name": "python3"
+    "c8d28cdf27fb43ddb41c83357775afcf": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
     },
-    "language_info": {
-      "name": "python"
+    "c93290da6a994795ae1e01e082cb2c66": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "DescriptionStyleModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "DescriptionStyleModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "StyleView",
+      "description_width": ""
+     }
     },
-    "widgets": {
-      "application/vnd.jupyter.widget-state+json": {
-        "03d9e42d6ecc4a269c5be231a0955f0f": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "ProgressStyleModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "ProgressStyleModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "bar_color": null,
-            "description_width": ""
-          }
-        },
-        "1518245edc554442bd0ad6804273309c": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "1976fec3079d4bac8c175c53f3f4ec32": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "HTMLModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "HTMLModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "HTMLView",
-            "description": "",
-            "description_tooltip": null,
-            "layout": "IPY_MODEL_b3b3c648b29e4019b83db9b1483416c2",
-            "placeholder": "​",
-            "style": "IPY_MODEL_c93290da6a994795ae1e01e082cb2c66",
-            "value": "Batches: 100%"
-          }
-        },
-        "3aa85f749d99477da52d8c3693046e6a": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "HTMLModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "HTMLModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "HTMLView",
-            "description": "",
-            "description_tooltip": null,
-            "layout": "IPY_MODEL_b4199cf8555a44cb80d22645750d0d32",
-            "placeholder": "​",
-            "style": "IPY_MODEL_9971ba728ebf44b2a2644c3930981ef4",
-            "value": " 1/1 [00:00&lt;00:00,  7.31it/s]"
-          }
-        },
-        "4ce46f0cb05641759d25207ced6cfe94": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "6f8fba6461ac4109b7bf6316967f316d": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "HTMLModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "HTMLModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "HTMLView",
-            "description": "",
-            "description_tooltip": null,
-            "layout": "IPY_MODEL_c8d28cdf27fb43ddb41c83357775afcf",
-            "placeholder": "​",
-            "style": "IPY_MODEL_c7183e1571594b049c4910a30c89be7f",
-            "value": " 1/1 [00:00&lt;00:00, 16.92it/s]"
-          }
-        },
-        "8208352647db4305bec1c2ab9499471b": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "DescriptionStyleModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "DescriptionStyleModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "description_width": ""
-          }
-        },
-        "836eef274ba14b01b4cad3a226a33d6f": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "HBoxModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "HBoxModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "HBoxView",
-            "box_style": "",
-            "children": [
-              "IPY_MODEL_1976fec3079d4bac8c175c53f3f4ec32",
-              "IPY_MODEL_b4fb8c6b9b424fa79560233caa9d8b40",
-              "IPY_MODEL_6f8fba6461ac4109b7bf6316967f316d"
-            ],
-            "layout": "IPY_MODEL_e110936ae62d471d8bb90ff1824ed9ad"
-          }
-        },
-        "9971ba728ebf44b2a2644c3930981ef4": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "DescriptionStyleModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "DescriptionStyleModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "description_width": ""
-          }
-        },
-        "a9d1bb3132a54b52a3f2f16aec5b94a7": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "b3b3c648b29e4019b83db9b1483416c2": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "b4199cf8555a44cb80d22645750d0d32": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "b4fb8c6b9b424fa79560233caa9d8b40": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "FloatProgressModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "FloatProgressModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "ProgressView",
-            "bar_style": "success",
-            "description": "",
-            "description_tooltip": null,
-            "layout": "IPY_MODEL_4ce46f0cb05641759d25207ced6cfe94",
-            "max": 1,
-            "min": 0,
-            "orientation": "horizontal",
-            "style": "IPY_MODEL_03d9e42d6ecc4a269c5be231a0955f0f",
-            "value": 1
-          }
-        },
-        "c7183e1571594b049c4910a30c89be7f": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "DescriptionStyleModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "DescriptionStyleModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "description_width": ""
-          }
-        },
-        "c8d28cdf27fb43ddb41c83357775afcf": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "c93290da6a994795ae1e01e082cb2c66": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "DescriptionStyleModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "DescriptionStyleModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "description_width": ""
-          }
-        },
-        "cc332ffcf5e5401ab26efef1b35fc34c": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "FloatProgressModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "FloatProgressModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "ProgressView",
-            "bar_style": "success",
-            "description": "",
-            "description_tooltip": null,
-            "layout": "IPY_MODEL_1518245edc554442bd0ad6804273309c",
-            "max": 1,
-            "min": 0,
-            "orientation": "horizontal",
-            "style": "IPY_MODEL_dadf5298941e41ad93e19aea51e3008f",
-            "value": 1
-          }
-        },
-        "ce61f00f6ff944d695002f7a8b599294": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "d441033160e4421bb9af18d941d50d6b": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "HBoxModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "HBoxModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "HBoxView",
-            "box_style": "",
-            "children": [
-              "IPY_MODEL_fb58298ecdf543389e615ccdf7e6a37a",
-              "IPY_MODEL_cc332ffcf5e5401ab26efef1b35fc34c",
-              "IPY_MODEL_3aa85f749d99477da52d8c3693046e6a"
-            ],
-            "layout": "IPY_MODEL_ce61f00f6ff944d695002f7a8b599294"
-          }
-        },
-        "dadf5298941e41ad93e19aea51e3008f": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "ProgressStyleModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "ProgressStyleModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "bar_color": null,
-            "description_width": ""
-          }
-        },
-        "e110936ae62d471d8bb90ff1824ed9ad": {
-          "model_module": "@jupyter-widgets/base",
-          "model_module_version": "1.2.0",
-          "model_name": "LayoutModel",
-          "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "LayoutView",
-            "align_content": null,
-            "align_items": null,
-            "align_self": null,
-            "border": null,
-            "bottom": null,
-            "display": null,
-            "flex": null,
-            "flex_flow": null,
-            "grid_area": null,
-            "grid_auto_columns": null,
-            "grid_auto_flow": null,
-            "grid_auto_rows": null,
-            "grid_column": null,
-            "grid_gap": null,
-            "grid_row": null,
-            "grid_template_areas": null,
-            "grid_template_columns": null,
-            "grid_template_rows": null,
-            "height": null,
-            "justify_content": null,
-            "justify_items": null,
-            "left": null,
-            "margin": null,
-            "max_height": null,
-            "max_width": null,
-            "min_height": null,
-            "min_width": null,
-            "object_fit": null,
-            "object_position": null,
-            "order": null,
-            "overflow": null,
-            "overflow_x": null,
-            "overflow_y": null,
-            "padding": null,
-            "right": null,
-            "top": null,
-            "visibility": null,
-            "width": null
-          }
-        },
-        "fb58298ecdf543389e615ccdf7e6a37a": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_module_version": "1.5.0",
-          "model_name": "HTMLModel",
-          "state": {
-            "_dom_classes": [],
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "HTMLModel",
-            "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "HTMLView",
-            "description": "",
-            "description_tooltip": null,
-            "layout": "IPY_MODEL_a9d1bb3132a54b52a3f2f16aec5b94a7",
-            "placeholder": "​",
-            "style": "IPY_MODEL_8208352647db4305bec1c2ab9499471b",
-            "value": "Batches: 100%"
-          }
-        }
-      }
+    "cc332ffcf5e5401ab26efef1b35fc34c": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "FloatProgressModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "FloatProgressModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "ProgressView",
+      "bar_style": "success",
+      "description": "",
+      "description_tooltip": null,
+      "layout": "IPY_MODEL_1518245edc554442bd0ad6804273309c",
+      "max": 1,
+      "min": 0,
+      "orientation": "horizontal",
+      "style": "IPY_MODEL_dadf5298941e41ad93e19aea51e3008f",
+      "value": 1
+     }
+    },
+    "ce61f00f6ff944d695002f7a8b599294": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
+    },
+    "d441033160e4421bb9af18d941d50d6b": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "HBoxModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "HBoxModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "HBoxView",
+      "box_style": "",
+      "children": [
+       "IPY_MODEL_fb58298ecdf543389e615ccdf7e6a37a",
+       "IPY_MODEL_cc332ffcf5e5401ab26efef1b35fc34c",
+       "IPY_MODEL_3aa85f749d99477da52d8c3693046e6a"
+      ],
+      "layout": "IPY_MODEL_ce61f00f6ff944d695002f7a8b599294"
+     }
+    },
+    "dadf5298941e41ad93e19aea51e3008f": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "ProgressStyleModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "ProgressStyleModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "StyleView",
+      "bar_color": null,
+      "description_width": ""
+     }
+    },
+    "e110936ae62d471d8bb90ff1824ed9ad": {
+     "model_module": "@jupyter-widgets/base",
+     "model_module_version": "1.2.0",
+     "model_name": "LayoutModel",
+     "state": {
+      "_model_module": "@jupyter-widgets/base",
+      "_model_module_version": "1.2.0",
+      "_model_name": "LayoutModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/base",
+      "_view_module_version": "1.2.0",
+      "_view_name": "LayoutView",
+      "align_content": null,
+      "align_items": null,
+      "align_self": null,
+      "border": null,
+      "bottom": null,
+      "display": null,
+      "flex": null,
+      "flex_flow": null,
+      "grid_area": null,
+      "grid_auto_columns": null,
+      "grid_auto_flow": null,
+      "grid_auto_rows": null,
+      "grid_column": null,
+      "grid_gap": null,
+      "grid_row": null,
+      "grid_template_areas": null,
+      "grid_template_columns": null,
+      "grid_template_rows": null,
+      "height": null,
+      "justify_content": null,
+      "justify_items": null,
+      "left": null,
+      "margin": null,
+      "max_height": null,
+      "max_width": null,
+      "min_height": null,
+      "min_width": null,
+      "object_fit": null,
+      "object_position": null,
+      "order": null,
+      "overflow": null,
+      "overflow_x": null,
+      "overflow_y": null,
+      "padding": null,
+      "right": null,
+      "top": null,
+      "visibility": null,
+      "width": null
+     }
+    },
+    "fb58298ecdf543389e615ccdf7e6a37a": {
+     "model_module": "@jupyter-widgets/controls",
+     "model_module_version": "1.5.0",
+     "model_name": "HTMLModel",
+     "state": {
+      "_dom_classes": [],
+      "_model_module": "@jupyter-widgets/controls",
+      "_model_module_version": "1.5.0",
+      "_model_name": "HTMLModel",
+      "_view_count": null,
+      "_view_module": "@jupyter-widgets/controls",
+      "_view_module_version": "1.5.0",
+      "_view_name": "HTMLView",
+      "description": "",
+      "description_tooltip": null,
+      "layout": "IPY_MODEL_a9d1bb3132a54b52a3f2f16aec5b94a7",
+      "placeholder": "​",
+      "style": "IPY_MODEL_8208352647db4305bec1c2ab9499471b",
+      "value": "Batches: 100%"
+     }
     }
-  },
-  "nbformat": 4,
-  "nbformat_minor": 0
+   }
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 0
 }