From b9cbc7627efa3612679aefa8ab5523eff6f0447c Mon Sep 17 00:00:00 2001
From: co-antwan <antoine@cohere.com>
Date: Tue, 12 Mar 2024 21:22:47 +0000
Subject: [PATCH 1/3] Adds notebooks with meeting notes rescipes

---
 ...pes_for_better_meeting_notes_summary.ipynb | 420 ++++++++++++++++++
 1 file changed, 420 insertions(+)
 create mode 100644 notebooks/Recipes_for_better_meeting_notes_summary.ipynb

diff --git a/notebooks/Recipes_for_better_meeting_notes_summary.ipynb b/notebooks/Recipes_for_better_meeting_notes_summary.ipynb
new file mode 100644
index 000000000..a709096a9
--- /dev/null
+++ b/notebooks/Recipes_for_better_meeting_notes_summary.ipynb
@@ -0,0 +1,420 @@
+{
+  "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "LplM9PSe8djM"
+      },
+      "source": [
+        "# Recipes for better meeting notes summaries\n",
+        "\n",
+        "This notebook builds on top of [our guide towards building better meeting summaries](https://colab.research.google.com/drive/1NuLdG6WmBfyedWR-pZIZBOH290ybL86T#scrollTo=YT6nsN6DK_xt). In that guide, we saw how to use Command-R to summarize meeting transcripts with a focus on imparting distinct formatting requirements.\n",
+        "\n",
+        "In this notebook, we'll cover useful recipes that we use internally to summarize our own meeting notes automatically (see [this guide](https://colab.research.google.com/drive/1BTAAV4ss-iPtxT0ueS7djbIIwPSt3Dpp) for an outline of the full system). We will focus on extracting specific items from the meeting notes, namely:\n",
+        "\n",
+        "1. Extract action items with assignees (who's on the hook for which task)\n",
+        "2. Summarise speaker perspectives (who said what)\n",
+        "3. Focus on a narrow topic (what was said about a given topic)\n",
+        "\n",
+        "Finally, we'll show that prompting the model for 1.-3. can be combined with the formatting instructions covered from our previous guide.\n",
+        "\n",
+        "We're constantly improving our summarisation capabilities across domains. For more information, you can reach out to us at summarize@cohere.com.\n",
+        "\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "6SfEhd3O74--"
+      },
+      "source": [
+        "## Setup\n",
+        "\n",
+        "You'll need a Cohere API key to run this notebook. If you don't have a key, head to https://cohere.com/ to generate your key."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "J1HzGqnY74bj",
+        "outputId": "e822cac4-debf-4f98-d238-0f3f104c44e9"
+      },
+      "outputs": [],
+      "source": [
+        "%%capture\n",
+        "!pip install cohere datasets tokenizers\n",
+        "\n",
+        "import cohere\n",
+        "from getpass import getpass\n",
+        "from datasets import load_dataset\n",
+        "\n",
+        "import re\n",
+        "from typing import Optional\n",
+        "\n",
+        "# Set up Cohere client\n",
+        "co_api_key = getpass(\"Enter your Cohere API key: \")\n",
+        "co_model = \"command-r\"\n",
+        "co = cohere.Client(co_api_key)\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "SbVCPxMSD3fs"
+      },
+      "outputs": [],
+      "source": [
+        "# We're also defining some util functions for later\n",
+        "\n",
+        "def pprint(s: Optional[str] = None, maxchars: int = 100):\n",
+        "  \"\"\"\n",
+        "  Wrap long text into lines of at most `maxchars` (preserves linebreaks occurring in text)\n",
+        "  \"\"\"\n",
+        "  if not s:\n",
+        "    print()\n",
+        "  else:\n",
+        "    print(\"\\n\".join(line.strip() for line in re.findall(rf\".{{1,{maxchars}}}(?:\\s+|$)\", s)))\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "ABgVdI7I8Jmk"
+      },
+      "source": [
+        "## Load test data\n",
+        "\n",
+        "Let's load a meeting transcript to see Command in action!\n",
+        "\n",
+        "* If you have your own transcript, you can load it to Colab using your favourite method.\n",
+        "* If you don't, we'll use a sample from the [QMSum dataset](https://github.com/Yale-LILY/QMSum). QMSum contains cleaned meeting transcripts with [diarised speakers](https://en.wikipedia.org/wiki/Speaker_diarisation); this will be perfect for testing our model's ability to assign action items to specific speakers.\n",
+        "* We'll see later that the recipe shared herein isn't limited to meeting notes transcript, but extends to any data with diarised speakers!"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "yS-hEiwQ71lv",
+        "outputId": "91343727-fef2-497f-fb98-2e0fbffb90fb"
+      },
+      "outputs": [],
+      "source": [
+        "# If you have your own transcript you would like to test Command on, load it here\n",
+        "# transcript = ...\n",
+        "\n",
+        "# Otherwise, we'll use QMSum\n",
+        "# Note this will download the QMSum dataset to your instance\n",
+        "qmsum = load_dataset(\"MocktaiLEngineer/qmsum-processed\")\n",
+        "# Pick any one transcript\n",
+        "transcripts = qmsum[\"validation\"][\"meeting_transcript\"]\n",
+        "transcript = transcripts[60]\n",
+        "pprint(transcript)\n",
+        "pprint()\n",
+        "\n",
+        "# Measure the number of tokens\n",
+        "num_tokens = len(co.tokenize(transcript).tokens)\n",
+        "pprint(f\"Number of tokens: {num_tokens}\")\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "TVWdxs3IIqpr"
+      },
+      "source": [
+        "If you've loaded the QMSum dataset, you'll see a back-and-forth dialogue between a Project Manager, an Industrial Designer, and two speakers responsible for Marketing and the User Interface. They seem engaged in a  design discussion with elements of retropsective. Let's see what action items followed from this meeting!\n",
+        "\n",
+        "Note that if your text length exceeds the context window of Command, you'll need to pre-process that text. Learn more about how to efficiently pre-process long texts at XXX [_include reference_].\n",
+        "\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "zSMTIfnH8xSk"
+      },
+      "source": [
+        "## Build the prompt to extract action items"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "kd3SCjWh9Zh9"
+      },
+      "source": [
+        "To extract action items, no special training is needed with Command!\n",
+        "\n",
+        "Here's one possible prompt to get action items with the speaker they were assigned to, in the form of bullet points:\n",
+        "\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "QAFIK0OP8vwU"
+      },
+      "outputs": [],
+      "source": [
+        "prompt_template = \"\"\"\n",
+        "## meeting transcript\n",
+        "{transcript}\n",
+        "\n",
+        "## instructions\n",
+        "Summarize the meeting transcript above, focusing it exclusively around action items. \\\n",
+        "Format your answer in the form of bullets. \\\n",
+        "Make sure to include the person each action item is assigned to. \\\n",
+        "Don't include preambles, postambles or explanations, but respond only with action items.\n",
+        "\n",
+        "## summary\n",
+        "\"\"\""
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "0PII_STgJ6oz"
+      },
+      "source": [
+        "We applied a few best practices in constructing this prompt:\n",
+        "\n",
+        "1. We include a preamble imbuing the model with a persona\n",
+        "2. We use Markdown-style headers (i.e. with `##`) to delineate the preamble, the text-to-summarise, the instructions, and the output\n",
+        "3. We make the instructions specific: we specify that we want action items with their assignees and specify the expected format\n",
+        "\n",
+        "Experiment with your own prompts, and let us know which worked best for you!"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "e2eK_CvN-5Ms"
+      },
+      "source": [
+        "## Generate action items\n",
+        "\n",
+        "Let's string together the prompt, and generate action items!"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "9zqeRjrI-3Jm",
+        "outputId": "a262bf9b-2980-4daf-bb47-aeec4b8fd68d"
+      },
+      "outputs": [],
+      "source": [
+        "prompt = prompt_template.format(transcript=transcript)\n",
+        "resp = co.chat(message=prompt, model=co_model, temperature=0.3)\n",
+        "action_items = resp.text\n",
+        "print(action_items)\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "e7taNqRTO2_A"
+      },
+      "source": [
+        "Not bad! We can see that the model successfully retrieved the action items that needed an immediate follow-up, and assigned the correct speaker to them. It also formatted the output as bullet points, as requested."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "anQARVO8RTMh"
+      },
+      "source": [
+        "Sometimes, we need to specify more complex output formats. For instance, we might want to automatically send action items to a company's project management software, which might only accept a certain data format.\n",
+        "\n",
+        "Say that software accepts only action items as JSON objects where assignees are keys. We could postprocess the previous response into that JSON. Or we could specify those requirements directly into our prompt template:"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "GYJ9CPD0ThYT",
+        "outputId": "b35e23e1-40e3-489f-a0d7-b1e70e059c71"
+      },
+      "outputs": [],
+      "source": [
+        "prompt_template_json = \"\"\"\n",
+        "## meeting transcript\n",
+        "{transcript}\n",
+        "\n",
+        "## instructions\n",
+        "Summarize the meeting transcript above, focusing it exclusively around action items. \\\n",
+        "Format your answer in the form of a compilable JSON, where every speaker is a new key. \\\n",
+        "Make sure to include the person each action item is assigned to. \\\n",
+        "Don't include preambles, postambles or explanations, but respond only with action items.\n",
+        "\n",
+        "## summary\n",
+        "\"\"\"\n",
+        "\n",
+        "prompt = prompt_template_json.format(transcript=transcript)\n",
+        "resp = co.chat(message=prompt, model=co_model, temperature=0.3)\n",
+        "action_items = resp.text\n",
+        "print(action_items)\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "DZ3X_kJB1mIM"
+      },
+      "source": [
+        "Great! All we needed was to specify the target output format for Command to obey.\n",
+        "\n",
+        "Try using different instructions to suit Command's outputs to your needs."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "igIW6yH_ip7_"
+      },
+      "source": [
+        "## How to replicate this pattern for more sub-tasks\n",
+        "\n",
+        "Command-R readily accommodates changes to the instructions to focus it on other subtasks. As an example, below we adapt the recipe developed for action items to perform two new tasks out-of-the-box:\n",
+        "\n",
+        "* a. Summarise the meeting from the point of view of every speaker\n",
+        "* b. Summarise everything's that been said about a specific topic"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "mw9esSWUjIE-"
+      },
+      "source": [
+        "#### a. User perspectives\n",
+        "\n",
+        "We'll also make the summary extractive, i.e. we encourage the model to reuse passages from the actual meeting."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "v_QSmS5rilHD",
+        "outputId": "6013098d-66e7-49ac-b43c-c0a28287e149"
+      },
+      "outputs": [],
+      "source": [
+        "prompt_template_perspectives = \"\"\"\n",
+        "## meeting transcript\n",
+        "{transcript}\n",
+        "\n",
+        "## instructions\n",
+        "Summarize the perspectives of every speaker. \\\n",
+        "Format your answer in the form of a JSON, where every speaker is a new key. Don't use your own words, but reuse passages from the meeting transcript where possible. \\\n",
+        "Don't include preambles, postambles or explanations, but respond only with your summary of each speaker's perspectives.\n",
+        "\n",
+        "## summary\n",
+        "\"\"\"\n",
+        "\n",
+        "prompt = prompt_template_perspectives.format(transcript=transcript)\n",
+        "resp = co.chat(message=prompt, model=co_model, temperature=0.3)\n",
+        "perspectives = resp.text\n",
+        "print(perspectives)\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "9ESmupp0jNoa"
+      },
+      "source": [
+        "#### b. Topic focus"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "X-JyBrpDilQ_",
+        "outputId": "a24edbe7-47ed-4e88-b8b0-ee4de45c8dba"
+      },
+      "outputs": [],
+      "source": [
+        "prompt_template_focus_topic = \"\"\"\\\n",
+        "## meeting transcript\n",
+        "{transcript}\n",
+        "\n",
+        "## instructions\n",
+        "Summarize everything that's been said about {topic} in the meeting transcript above. If the meeting transcript doesn't mention {topic}, simply state \\\n",
+        "that there is no trace of {topic} in the provided meeting transcript. \\\n",
+        "Format your answer in the form of paragraphs. \\\n",
+        "Don't include preambles, postambles or explanations, but respond only with your summary of {topic}.\n",
+        "\n",
+        "## summary\n",
+        "\"\"\"\n",
+        "\n",
+        "# Try new topics here!\n",
+        "topic = \"objects shaped like bananas\"\n",
+        "\n",
+        "prompt = prompt_template_focus_topic.format(transcript=transcript, topic=topic)\n",
+        "resp = co.chat(message=prompt, model=co_model, temperature=0.3)\n",
+        "perspectives = resp.text\n",
+        "print(perspectives)\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "0ytrkOUYjPXm"
+      },
+      "source": [
+        "Try some of your own prompts and share them back with us at summarize@cohere.com!"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "dOlPsA2Q4q_y"
+      },
+      "outputs": [],
+      "source": []
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}

From 6ef67dd06cb447e3dbdc3ea4e61320439d910cea Mon Sep 17 00:00:00 2001
From: boyu-cohere <boyu@cohere.com>
Date: Tue, 12 Mar 2024 21:48:03 -0400
Subject: [PATCH 2/3] added general meeting notes notebook

---
 ...eeting_Summaries_General_&_LangChain.ipynb | 3128 +++++++++++++++++
 1 file changed, 3128 insertions(+)
 create mode 100644 notebooks/Meeting_Summaries_General_&_LangChain.ipynb

diff --git a/notebooks/Meeting_Summaries_General_&_LangChain.ipynb b/notebooks/Meeting_Summaries_General_&_LangChain.ipynb
new file mode 100644
index 000000000..e6d1d2e94
--- /dev/null
+++ b/notebooks/Meeting_Summaries_General_&_LangChain.ipynb
@@ -0,0 +1,3128 @@
+{
+  "cells": [
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "LplM9PSe8djM"
+      },
+      "source": [
+        "# Summarizing meeting notes\n",
+        "\n",
+        "Welcome to Cohere! In this notebook, you'll learn how to:\n",
+        "* Use Cohere's Command-R model to summarize meeting transcripts.\n",
+        "* Modify your prompt to include specific formatting instructions, especially if the model output is used in downstream applications.\n",
+        "* Use Command-R with LangChain for summarization."
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "6SfEhd3O74--"
+      },
+      "source": [
+        "## Setup\n",
+        "\n",
+        "You'll need a Cohere API key to run this notebook. If you don't have a key, head to https://cohere.com/ to generate your key."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 1,
+      "metadata": {
+        "id": "H2TfKiPM3a6f"
+      },
+      "outputs": [],
+      "source": [
+        "%%capture\n",
+        "!pip install cohere datasets tokenizers langchain"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 2,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "J1HzGqnY74bj",
+        "outputId": "24fba7c5-a2ce-4c86-bc14-4ca2fb9137b5"
+      },
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "Enter your Cohere API key: ··········\n"
+          ]
+        }
+      ],
+      "source": [
+        "import re\n",
+        "from string import Template\n",
+        "from typing import Optional\n",
+        "\n",
+        "import cohere\n",
+        "from getpass import getpass\n",
+        "from datasets import load_dataset\n",
+        "\n",
+        "# Set up Cohere client\n",
+        "co_api_key = getpass(\"Enter your Cohere API key: \")\n",
+        "co_model = \"command-r\"\n",
+        "co = cohere.Client(api_key=co_api_key)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 3,
+      "metadata": {
+        "id": "SbVCPxMSD3fs"
+      },
+      "outputs": [],
+      "source": [
+        "# We'll also defining some util functions for later\n",
+        "\n",
+        "def pprint(s: Optional[str] = None, maxchars: int = 100):\n",
+        "  \"\"\"\n",
+        "  Wrap long text into lines of at most `maxchars` (preserves linebreaks occurring in text)\n",
+        "  \"\"\"\n",
+        "  if not s:\n",
+        "    print()\n",
+        "  else:\n",
+        "    print(\"\\n\".join(line.strip() for line in re.findall(rf\".{{1,{maxchars}}}(?:\\s+|$)\", s)))\n"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "ABgVdI7I8Jmk"
+      },
+      "source": [
+        "## Load test data\n",
+        "\n",
+        "Let's load a meeting transcript to see Command in action!\n",
+        "\n",
+        "* If you have your own transcript, you can load it to Colab using your favorite method.\n",
+        "* If you don't, we'll use a sample from the [QMSum dataset](https://github.com/Yale-LILY/QMSum). QMSum contains cleaned meeting transcripts with [diarised speakers](https://en.wikipedia.org/wiki/Speaker_diarisation).\n",
+        "* We'll see later that the recipe shared herein isn't limited to meeting notes transcript, but extends to any data with diarised speakers!"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 4,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 1000,
+          "referenced_widgets": [
+            "43225bc344ed4374931511b8630baae4",
+            "56d2b7a171c4497a83e26bbafbbb4612",
+            "2f73ed56c08a45c9ab1dbdfb59eb51cd",
+            "4798a0f8e4d24a568fded08cde65dc7a",
+            "68aa82f9c86f47b6a810aa0404bb8266",
+            "09fa50ba56b6483d88413799d45e6baa",
+            "079f8238a7124707963f240aaef7eb4f",
+            "904ddfbc8af44e43bb32a7c14639de8d",
+            "38a4dabfd7b648dca78f2d6159079779",
+            "7922f387c6a04bfc97bdd046097b567d",
+            "95d74166787244d4807e5c4b0223ab4c",
+            "23801508bc9c414cb106b6809c5f3043",
+            "2e1ca43dfaf14957ad788519f57333e9",
+            "9df7c90e2a9444c1b97dc055bb805a4e",
+            "5f9d1e97dd62416587abfc724743c685",
+            "6b68bf1fc952421c8534b262dc6b27b9",
+            "5f2d4056e0be4205bd09948715a66f07",
+            "5f3a09f953e64b44bb91a8258d91ef28",
+            "5521d0c2117a4500b741197c99da7d36",
+            "99872e6b03a047a38b1ea88976c4b1a4",
+            "a4bc752fd44a4a0da969d08fd232644a",
+            "cc7e4555f04f4d09aab780af5dce5004",
+            "0d41cdf4e8b049d48912ee8ae20e5bf4",
+            "dc022d95fff345e88fe3fdeb42ad78c8",
+            "73fa502143aa4186a32bcf58fdfe0424",
+            "88abb7063a4f4f43b48e51ba746ec6e6",
+            "a2f91869ede34418ad6d934680a4a447",
+            "f1a3b77843af4b219e24f94fe7224516",
+            "5895c33db870460681eb05f98da67a5b",
+            "8ebf85a7dd4c4a538405100fb08c7376",
+            "52b2f525cfdb429aa79cd7097989b657",
+            "fef8c84b911a419c882ae7c0d8b6a937",
+            "9cecfc3ecc2c4027a76182fb4330ea6a",
+            "fd57fbd3688942ae9a4a2c0c65817c88",
+            "af362384efb04c50bd2e3d16abb28448",
+            "671324a8ab0249a59e9d10f7b7df0d3a",
+            "2e9d71ebf60049a68dce99ae57633ed6",
+            "3eaf53e3d6674527aaca6f4aeda71f7d",
+            "e829f283ba864af1af50cdce847692c1",
+            "0ce661f36e724d24a65d875106b3e92d",
+            "0cdd8ccc3bef493ba59b3ca0cae5f785",
+            "6ed77172d5da420093304ccde9420f98",
+            "b9284cee75b24777815d37512975fa64",
+            "e8e40b743abc4c0d8f0fcc66a2d45c30",
+            "6ed3be205c90420bbe4a2ff7911ef4ec",
+            "28df182d45c54455974d58435ea5407f",
+            "dff1f82298b0499f92ac781678245018",
+            "8530897eee1f4512b95617d5e81c890d",
+            "266ee16a3cda44bfb634366770e52ca2",
+            "074a041d836046219e201ab8443e994d",
+            "3973b1ea9bdd42319bea13839b04ff73",
+            "ac0960ad1e8b42f29fe0342afe9a5dea",
+            "36bd2c9c60884f6fac47d99ac8e62df7",
+            "35b591964bbd436da1eed4cb18063085",
+            "97e38497aedd4e52a2969a75f40e69cf",
+            "f63f4d33706a4a6bb275e895cce1399a",
+            "cc16e3c53da9411aa5e705cb827a1ee0",
+            "1b415516e0b74d70bed24553c562652e",
+            "0e0f7993c16b44689184af9c92420ab7",
+            "073a3fe3ace242fb96da7338a8769316",
+            "d5d56db4f0394ec0baf682d3367d1750",
+            "9698cc34f7114de58518334c1e4acc79",
+            "dc8e7b2d8bbf44e5b8d94c9f4641928b",
+            "41f87857e69641c7bd469f9305fbb77f",
+            "ae4f97637ec84bf786780865cb7590aa",
+            "5c4d39b44cce45debd1c79320f99265a",
+            "918da148fd1d42cd80e890600d7a7de9",
+            "81b7e458c813477fbf2994ae8a5202ba",
+            "f1448623a7074e29af9b7bdcfd1ba52c",
+            "100aa10071bf49098bc82ffcf34be553",
+            "6e9f3a291822403a8e25c243edca3d6a",
+            "4a69736428824330bc41a08fa8f42684",
+            "582d77ff31aa48d28913f8d19fa402a7",
+            "a30cc578ba9c48f4b046731fb47df0af",
+            "5f8e24fef9fc459db426d955f002a764",
+            "2b04db4915fc4c7fa54a6ff0e5efed03",
+            "9c8a77cd6bce4d9a8e3c81714cad6964"
+          ]
+        },
+        "id": "yS-hEiwQ71lv",
+        "outputId": "30c66020-c5b0-4984-fce6-9ae018b7abf0"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stderr",
+          "text": [
+            "/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_token.py:88: UserWarning: \n",
+            "The secret `HF_TOKEN` does not exist in your Colab secrets.\n",
+            "To authenticate with the Hugging Face Hub, create a token in your settings tab (https://huggingface.co/settings/tokens), set it as secret in your Google Colab and restart your session.\n",
+            "You will be able to reuse this secret in all of your notebooks.\n",
+            "Please note that authentication is recommended but still optional to access public models or datasets.\n",
+            "  warnings.warn(\n"
+          ]
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading readme:   0%|          | 0.00/21.0 [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "43225bc344ed4374931511b8630baae4"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/4.45M [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "23801508bc9c414cb106b6809c5f3043"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/950k [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "0d41cdf4e8b049d48912ee8ae20e5bf4"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/917k [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "fd57fbd3688942ae9a4a2c0c65817c88"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Generating train split: 0 examples [00:00, ? examples/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "6ed3be205c90420bbe4a2ff7911ef4ec"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Generating validation split: 0 examples [00:00, ? examples/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "f63f4d33706a4a6bb275e895cce1399a"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Generating test split: 0 examples [00:00, ? examples/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "918da148fd1d42cd80e890600d7a7de9"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "PhD D: Sure because I I need a lot of time to to put the label or to do that\n",
+            "Professor E: I mean we we know that there s noise There s there s continual noise from fans and so\n",
+            "forth and there is more impulsive noise from taps and so forth and and something in between with\n",
+            "paper rustling We know that all that s there and it s a g worthwhile thing to study but obviously it\n",
+            "takes a lot of time to mark all of these things Whereas th i I would think that you we can study\n",
+            "more or less as a distinct phenomenon the overlapping of people talking So Then you can get the Cuz\n",
+            "you need If it s three hundred i i it sounds like you probably only have fifty or sixty or seventy\n",
+            "events right now that are really And and you need to have a lot more than that to have any kind of\n",
+            "even visual sense of of what s going on much less any kind of reasonable statistics\n",
+            "PhD C: Now why do you need to mark speaker overlap by hand if you can infer it from the relative\n",
+            "energy in the\n",
+            "Grad G: Well that s That s what I was going to bring up\n",
+            "PhD C: I mean you should not need to do this p completely by hand\n",
+            "Professor E: OK So let s back up because you were not here for an earlier conversation\n",
+            "PhD C: right ? I m sorry\n",
+            "Professor E: So the idea was that what he was going to be doing was experimenting with different\n",
+            "measures such as the increase in energy such as the energy in the LPC residuals such as I mean there\n",
+            "s a bunch of things I mean increased energy is is sort of an obvious one\n",
+            "PhD C: Mm In the far field mike\n",
+            "Professor E: and it s not obvious I mean you could you could do the dumbest thing and get get it\n",
+            "ninety percent of the time But when you start going past that and trying to do better it s not\n",
+            "obvious what combination of features is going to give you the you know the right detector So the\n",
+            "idea is to have some ground truth first And so the i the idea of the manual marking was to say `` OK\n",
+            "this i you know it s it s really here ``\n",
+            "PhD A: But I think Liz is saying why not get it out of the transcripts ?\n",
+            "PhD C: What I mean is pause get it from the close talking mikes A or ge get a first pass from those\n",
+            "Professor E: We t we t w we t we talked about that\n",
+            "PhD C: and then go through sort of It would be a lot faster probably to\n",
+            "Professor E: We we we talked about that s But so it s a bootstrapping thing and the thing is the\n",
+            "idea was i we i i we thought it would be useful for him to look at the data anyway and and then\n",
+            "whatever he could mark would be helpful and we could it s a question of what you bootstrap from You\n",
+            "know do you bootstrap from a simple measurement which is right most of the time and then you g do\n",
+            "better or do you bootstrap from some human being looking at it and then then do your simple\n",
+            "measurements from the close talking mike I mean even with the close talking mike you are not going\n",
+            "to get it right all the time\n",
+            "PhD C: Well that s what I wonder because or how bad it is be because that would be interesting\n",
+            "Grad G: I m working on a program to do that and\n",
+            "PhD C: especially because the bottleneck is the transcription Right ? I mean we ve got a lot more\n",
+            "data than we have transcriptions for We have the audio data we have the close talking mike so I mean\n",
+            "it seems like one kind of project that s not perfect but that you can get the training data for\n",
+            "pretty quickly is you know if you infer form the close talking mikes where the on off points are of\n",
+            "speech you know how can we detect that from a far field ?\n",
+            "Grad G: I ve I ve written a program to do that\n",
+            "PhD C: OK I m sorry I missed the\n",
+            "Grad G: and so but it s it s doing something very very simple It just takes a threshold based on on\n",
+            "the volume\n",
+            "PhD F: Or you can set the threshold low and then weed out the false alarms by hand\n",
+            "Grad G: and then it does a median filter and then it looks for runs And it seems to work I ve I m\n",
+            "sort of fiddling with the parameters to get it to actually generate something and I have not I do\n",
+            "not what I m working on was working on was getting it to a form where we can import it into the user\n",
+            "interface that we have pause into Transcriber And so I told I said it would take about a day I ve\n",
+            "worked on it for about half a day so give me another half day and I we will have something we can\n",
+            "play with\n",
+            "Professor E: See this is where we really need the Meeting Recorder query stuff to be working because\n",
+            "we ve had these meetings and we ve had this discussion about this and I m sort of remembering a\n",
+            "little bit about what we decided\n",
+            "PhD C: Right I m sorry I just\n",
+            "Professor E: but I could not remember all of it So I think it was partly that you know give somebody\n",
+            "a chance to actually look at the data and see what these are like partly that we have e some ground\n",
+            "truth to compare against you know when when he he gets his thing going\n",
+            "PhD C: Well it s definitely good to have somebody look at it I was just thinking as a way to speed\n",
+            "up you know the amount of\n",
+            "Professor E: That was that was exactly the notion that that that we discussed\n"
+          ]
+        }
+      ],
+      "source": [
+        "# If you have your own transcript you would like to test Command on, load it here\n",
+        "# transcript = ...\n",
+        "\n",
+        "# Otherwise, we'll use QMSum\n",
+        "# Note this will download the QMSum dataset to your instance\n",
+        "qmsum = load_dataset(\"MocktaiLEngineer/qmsum-processed\")\n",
+        "# Pick any one transcript\n",
+        "transcripts = qmsum[\"validation\"][\"meeting_transcript\"]\n",
+        "transcript = transcripts[0]\n",
+        "pprint(transcript)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "TVWdxs3IIqpr"
+      },
+      "source": [
+        "If you've loaded the QMSum dataset, you'll see a back-and-forth dialogue between a Project Manager, an Industrial Designer, and someone responsible for the User Interface. They discuss the design of a banana-shaped remote controller.\n",
+        "\n",
+        "Let's also check how many tokens that transcript corresponds to."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# Measure the number of tokens\n",
+        "num_tokens = len(co.tokenize(transcript, model=co_model).tokens)\n",
+        "pprint(f\"Number of tokens: {num_tokens}\")"
+      ],
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "eGlS8mMX5Kdw",
+        "outputId": "844aa95f-8a81-48e0-c8fd-6dea58bd33c7"
+      },
+      "execution_count": 5,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Number of tokens: 1152\n"
+          ]
+        }
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "zSMTIfnH8xSk"
+      },
+      "source": [
+        "## Simple summarization\n",
+        "\n",
+        "First, we'll show an example of a simple summarization task. The prompt template we are using is shown below, and we can obtain the completion using a one line call to the Cohere API."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 6,
+      "metadata": {
+        "id": "QAFIK0OP8vwU",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "37300916-0f32-4826-9bcf-608142ce0676"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "The team discusses the time-consuming process of identifying instances of speaker overlap in audio\n",
+            "recordings. PhD C suggests using close-talking microphones and automated methods to establish ground\n",
+            "truth data, which Grad G is working on. This data would help develop a more efficient system than\n",
+            "manual marking.\n"
+          ]
+        }
+      ],
+      "source": [
+        "prompt_template = Template(\"\"\"## meeting transcript\n",
+        "$transcript\n",
+        "\n",
+        "## instructions\n",
+        "Generate a summary of the above meeting transcript. \\\n",
+        "Don't include preambles, postambles or explanations. \\\n",
+        "\"\"\")\n",
+        "prompt = prompt_template.substitute(transcript=transcript)\n",
+        "\n",
+        "# Use the Cohere API to get the response\n",
+        "resp = co.chat(message=prompt, model=co_model, temperature=0.2).text\n",
+        "pprint(resp)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "0PII_STgJ6oz"
+      },
+      "source": [
+        "We applied a few best practices in constructing this prompt:\n",
+        "\n",
+        "1. We include a preamble imbuing the model with a persona.\n",
+        "2. We use Markdown-style headers (i.e. with `##`) to delineate the preamble, the text-to-summarise, the instructions, and the output\n",
+        "\n",
+        "For some other prompting best practices and tips/tricks, see our [prompting guide](https://docs.google.com/document/d/1k0KBiOUEPq65vqEuAQyg6Y5l1s2_Ybwv-lOLho6NHsw/edit)!"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "YT6nsN6DK_xt"
+      },
+      "source": [
+        "## More advanced formatting\n",
+        "\n",
+        "That worked well, but what if you want to have more control over the summary? What if you wanted a bulleted summary and wanted exactly 3 bullets in a json structure?\n",
+        "\n",
+        "Let's try adding in some more specific formatting instructions.\n"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 7,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "hBw5GnX7K_xt",
+        "outputId": "7bba22f3-3618-4e62-b010-8c55b96188b6"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "```json\n",
+            "{\n",
+            "    \"summary_bullets\": [\n",
+            "        \"There's a need to identify instances of speaker overlap in recorded meetings, but it's a time-consuming task to mark these events manually.\",\n",
+            "        \"An alternative is to infer speaker overlap using various measurements of energy in the audio, such as close-talking microphones, LPC residuals, etc.\",\n",
+            "        \"A threshold-based program that detects speech based on volume can speed up the process, and further improvements can be explored once a robust system is in place.\"\n",
+            "    ]\n",
+            "}\n",
+            "```\n"
+          ]
+        }
+      ],
+      "source": [
+        "prompt_template = Template(\"\"\"## meeting transcript\n",
+        "$transcript\n",
+        "\n",
+        "## instructions\n",
+        "Generate a summary of the above meeting transcript in 3 bullets. \\\n",
+        "Use the following JSON format:\n",
+        "{\n",
+        "    \"summary_bullets\": [\n",
+        "        \"<bullet 1>\",\n",
+        "        \"<bullet 2>\",\n",
+        "        ...\n",
+        "    ]\n",
+        "}\n",
+        "Don't include preambles, postambles or explanations.\\\n",
+        "\"\"\")\n",
+        "prompt = prompt_template.substitute(transcript=transcript)\n",
+        "\n",
+        "# Use the Cohere API to get the response\n",
+        "resp = co.chat(message=prompt, model=co_model, temperature=0.2).text\n",
+        "print(resp)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "That worked well! Asking for a structured response like a JSON object is a great approach for use cases where you will need to parse or post-process the output for use in downstream applications. To ensure the best possible completion, I provided a template of the expected JSON that the model filled out.\n",
+        "\n",
+        "Let's also try asking for a short 1 sentence summary."
+      ],
+      "metadata": {
+        "id": "inBjpwOK6pTN"
+      }
+    },
+    {
+      "cell_type": "code",
+      "execution_count": 8,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "y5aC_E4fK_xt",
+        "outputId": "aeefd38d-4673-443b-f959-f500e281b764"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "The team discusses using various methods, including close-talking mics and threshold-based volume\n",
+            "filters, to accurately identify instances of speaker overlap in recorded meetings.\n"
+          ]
+        }
+      ],
+      "source": [
+        "prompt_template = Template(\"\"\"## meeting transcript\n",
+        "$transcript\n",
+        "\n",
+        "## instructions\n",
+        "Generate a summary of the above meeting transcript in 1 concise sentence. Make sure the sentence is extremely short.\\\n",
+        "Don't include preambles, postambles or explanations.\\\n",
+        "\"\"\")\n",
+        "prompt = prompt_template.substitute(transcript=transcript)\n",
+        "\n",
+        "# Use the Cohere API to get the response\n",
+        "resp = co.chat(message=prompt, model=co_model, temperature=0.2).text\n",
+        "pprint(resp)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "## LangChain summarization\n",
+        "It's also very easy to use Command-R with LangChain for summarization."
+      ],
+      "metadata": {
+        "id": "KOaKiT6Ztkbz"
+      }
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "from langchain.chains.combine_documents.stuff import StuffDocumentsChain\n",
+        "from langchain.chains.llm import LLMChain\n",
+        "from langchain.prompts import PromptTemplate\n",
+        "from langchain.docstore.document import Document\n",
+        "from langchain_community.chat_models import ChatCohere\n",
+        "\n",
+        "llm = ChatCohere(\n",
+        "    cohere_api_key=co_api_key,\n",
+        "    model=\"command-r\",\n",
+        "    temperature=0.2,\n",
+        "  )\n",
+        "\n",
+        "# Define prompt\n",
+        "prompt_template = \"\"\"## meeting transcript\n",
+        "{transcript}\n",
+        "\n",
+        "## instructions\n",
+        "Generate a summary of the above meeting transcript. \\\n",
+        "Don't include preambles, postambles or explanations.\"\"\"\n",
+        "prompt = PromptTemplate.from_template(prompt_template)\n",
+        "\n",
+        "# Define LLM chain\n",
+        "llm_chain = LLMChain(llm=llm, prompt=prompt)\n",
+        "\n",
+        "# Define StuffDocumentsChain\n",
+        "stuff_chain = StuffDocumentsChain(llm_chain=llm_chain, document_variable_name=\"transcript\", verbose=True)\n",
+        "\n",
+        "# Load documents\n",
+        "docs = [Document(page_content=transcript)]\n",
+        "\n",
+        "pprint(stuff_chain.run(docs))"
+      ],
+      "metadata": {
+        "id": "5eJM29wWtx-m",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "3f29b633-e406-4608-95d3-60653ff41c51"
+      },
+      "execution_count": 9,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stderr",
+          "text": [
+            "/usr/local/lib/python3.10/dist-packages/langchain_core/_api/deprecation.py:117: LangChainDeprecationWarning: The function `run` was deprecated in LangChain 0.1.0 and will be removed in 0.2.0. Use invoke instead.\n",
+            "  warn_deprecated(\n"
+          ]
+        },
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "\n",
+            "\n",
+            "\u001b[1m> Entering new StuffDocumentsChain chain...\u001b[0m\n",
+            "\n",
+            "\u001b[1m> Finished chain.\u001b[0m\n",
+            "The team discusses the time-consuming process of identifying instances of speaker overlap in audio\n",
+            "recordings. PhD C suggests using close-talking microphones and automated methods to establish ground\n",
+            "truth data, which Grad G is working on. This data would help develop a more efficient system than\n",
+            "manual marking.\n"
+          ]
+        }
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "Command-R has a context window of 128K tokens, so you can use very long documents, or multiple documents, all in a single API call.\n",
+        "\n"
+      ],
+      "metadata": {
+        "id": "MtS7tZi9t-dY"
+      }
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "## Wrap up\n",
+        "\n",
+        "Those were some simple examples of how you can use Cohere's Command-R model for summarization.\n",
+        "\n",
+        "For more advanced summarization objectives and to see some other cool things you can do with the Command-R model, see [recipes for better meeting notes summaries](https://colab.research.google.com/drive/1XqRpJH7qRnRTbOEt6kthwqZG6gtEn4gN#scrollTo=LplM9PSe8djM).\n",
+        "\n",
+        "In cases where you can't fit the input text into the context window, see our [cookbook for long document strategies](https://colab.research.google.com/drive/1zxSAbruOWwWJHNsj3N56uxZtUeiS7Evd#scrollTo=4JgYYTg7Shvq).\n",
+        "\n",
+        "For more information, you can reach out to us at summarize@cohere.com!"
+      ],
+      "metadata": {
+        "id": "uj8kdy5auKKY"
+      }
+    }
+  ],
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "codemirror_mode": {
+        "name": "ipython",
+        "version": 3
+      },
+      "file_extension": ".py",
+      "mimetype": "text/x-python",
+      "name": "python",
+      "nbconvert_exporter": "python",
+      "pygments_lexer": "ipython3",
+      "version": "3.10.11"
+    },
+    "widgets": {
+      "application/vnd.jupyter.widget-state+json": {
+        "43225bc344ed4374931511b8630baae4": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HBoxModel",
+          "model_module_version": "1.5.0",
+          "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_56d2b7a171c4497a83e26bbafbbb4612",
+              "IPY_MODEL_2f73ed56c08a45c9ab1dbdfb59eb51cd",
+              "IPY_MODEL_4798a0f8e4d24a568fded08cde65dc7a"
+            ],
+            "layout": "IPY_MODEL_68aa82f9c86f47b6a810aa0404bb8266"
+          }
+        },
+        "56d2b7a171c4497a83e26bbafbbb4612": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_09fa50ba56b6483d88413799d45e6baa",
+            "placeholder": "​",
+            "style": "IPY_MODEL_079f8238a7124707963f240aaef7eb4f",
+            "value": "Downloading readme: 100%"
+          }
+        },
+        "2f73ed56c08a45c9ab1dbdfb59eb51cd": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "FloatProgressModel",
+          "model_module_version": "1.5.0",
+          "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_904ddfbc8af44e43bb32a7c14639de8d",
+            "max": 21,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_38a4dabfd7b648dca78f2d6159079779",
+            "value": 21
+          }
+        },
+        "4798a0f8e4d24a568fded08cde65dc7a": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_7922f387c6a04bfc97bdd046097b567d",
+            "placeholder": "​",
+            "style": "IPY_MODEL_95d74166787244d4807e5c4b0223ab4c",
+            "value": " 21.0/21.0 [00:00&lt;00:00, 767B/s]"
+          }
+        },
+        "68aa82f9c86f47b6a810aa0404bb8266": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "09fa50ba56b6483d88413799d45e6baa": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "079f8238a7124707963f240aaef7eb4f": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "904ddfbc8af44e43bb32a7c14639de8d": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "38a4dabfd7b648dca78f2d6159079779": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ProgressStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "7922f387c6a04bfc97bdd046097b567d": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "95d74166787244d4807e5c4b0223ab4c": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "23801508bc9c414cb106b6809c5f3043": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HBoxModel",
+          "model_module_version": "1.5.0",
+          "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_2e1ca43dfaf14957ad788519f57333e9",
+              "IPY_MODEL_9df7c90e2a9444c1b97dc055bb805a4e",
+              "IPY_MODEL_5f9d1e97dd62416587abfc724743c685"
+            ],
+            "layout": "IPY_MODEL_6b68bf1fc952421c8534b262dc6b27b9"
+          }
+        },
+        "2e1ca43dfaf14957ad788519f57333e9": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_5f2d4056e0be4205bd09948715a66f07",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5f3a09f953e64b44bb91a8258d91ef28",
+            "value": "Downloading data: 100%"
+          }
+        },
+        "9df7c90e2a9444c1b97dc055bb805a4e": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "FloatProgressModel",
+          "model_module_version": "1.5.0",
+          "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_5521d0c2117a4500b741197c99da7d36",
+            "max": 4445270,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_99872e6b03a047a38b1ea88976c4b1a4",
+            "value": 4445270
+          }
+        },
+        "5f9d1e97dd62416587abfc724743c685": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_a4bc752fd44a4a0da969d08fd232644a",
+            "placeholder": "​",
+            "style": "IPY_MODEL_cc7e4555f04f4d09aab780af5dce5004",
+            "value": " 4.45M/4.45M [00:00&lt;00:00, 5.31MB/s]"
+          }
+        },
+        "6b68bf1fc952421c8534b262dc6b27b9": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "5f2d4056e0be4205bd09948715a66f07": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "5f3a09f953e64b44bb91a8258d91ef28": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "5521d0c2117a4500b741197c99da7d36": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "99872e6b03a047a38b1ea88976c4b1a4": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ProgressStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "a4bc752fd44a4a0da969d08fd232644a": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "cc7e4555f04f4d09aab780af5dce5004": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "0d41cdf4e8b049d48912ee8ae20e5bf4": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HBoxModel",
+          "model_module_version": "1.5.0",
+          "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_dc022d95fff345e88fe3fdeb42ad78c8",
+              "IPY_MODEL_73fa502143aa4186a32bcf58fdfe0424",
+              "IPY_MODEL_88abb7063a4f4f43b48e51ba746ec6e6"
+            ],
+            "layout": "IPY_MODEL_a2f91869ede34418ad6d934680a4a447"
+          }
+        },
+        "dc022d95fff345e88fe3fdeb42ad78c8": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_f1a3b77843af4b219e24f94fe7224516",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5895c33db870460681eb05f98da67a5b",
+            "value": "Downloading data: 100%"
+          }
+        },
+        "73fa502143aa4186a32bcf58fdfe0424": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "FloatProgressModel",
+          "model_module_version": "1.5.0",
+          "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_8ebf85a7dd4c4a538405100fb08c7376",
+            "max": 950100,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_52b2f525cfdb429aa79cd7097989b657",
+            "value": 950100
+          }
+        },
+        "88abb7063a4f4f43b48e51ba746ec6e6": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_fef8c84b911a419c882ae7c0d8b6a937",
+            "placeholder": "​",
+            "style": "IPY_MODEL_9cecfc3ecc2c4027a76182fb4330ea6a",
+            "value": " 950k/950k [00:00&lt;00:00, 1.80MB/s]"
+          }
+        },
+        "a2f91869ede34418ad6d934680a4a447": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "f1a3b77843af4b219e24f94fe7224516": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "5895c33db870460681eb05f98da67a5b": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "8ebf85a7dd4c4a538405100fb08c7376": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "52b2f525cfdb429aa79cd7097989b657": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ProgressStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "fef8c84b911a419c882ae7c0d8b6a937": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "9cecfc3ecc2c4027a76182fb4330ea6a": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "fd57fbd3688942ae9a4a2c0c65817c88": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HBoxModel",
+          "model_module_version": "1.5.0",
+          "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_af362384efb04c50bd2e3d16abb28448",
+              "IPY_MODEL_671324a8ab0249a59e9d10f7b7df0d3a",
+              "IPY_MODEL_2e9d71ebf60049a68dce99ae57633ed6"
+            ],
+            "layout": "IPY_MODEL_3eaf53e3d6674527aaca6f4aeda71f7d"
+          }
+        },
+        "af362384efb04c50bd2e3d16abb28448": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_e829f283ba864af1af50cdce847692c1",
+            "placeholder": "​",
+            "style": "IPY_MODEL_0ce661f36e724d24a65d875106b3e92d",
+            "value": "Downloading data: 100%"
+          }
+        },
+        "671324a8ab0249a59e9d10f7b7df0d3a": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "FloatProgressModel",
+          "model_module_version": "1.5.0",
+          "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_0cdd8ccc3bef493ba59b3ca0cae5f785",
+            "max": 916846,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_6ed77172d5da420093304ccde9420f98",
+            "value": 916846
+          }
+        },
+        "2e9d71ebf60049a68dce99ae57633ed6": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_b9284cee75b24777815d37512975fa64",
+            "placeholder": "​",
+            "style": "IPY_MODEL_e8e40b743abc4c0d8f0fcc66a2d45c30",
+            "value": " 917k/917k [00:00&lt;00:00, 1.64MB/s]"
+          }
+        },
+        "3eaf53e3d6674527aaca6f4aeda71f7d": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "e829f283ba864af1af50cdce847692c1": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "0ce661f36e724d24a65d875106b3e92d": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "0cdd8ccc3bef493ba59b3ca0cae5f785": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "6ed77172d5da420093304ccde9420f98": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ProgressStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "b9284cee75b24777815d37512975fa64": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "e8e40b743abc4c0d8f0fcc66a2d45c30": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "6ed3be205c90420bbe4a2ff7911ef4ec": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HBoxModel",
+          "model_module_version": "1.5.0",
+          "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_28df182d45c54455974d58435ea5407f",
+              "IPY_MODEL_dff1f82298b0499f92ac781678245018",
+              "IPY_MODEL_8530897eee1f4512b95617d5e81c890d"
+            ],
+            "layout": "IPY_MODEL_266ee16a3cda44bfb634366770e52ca2"
+          }
+        },
+        "28df182d45c54455974d58435ea5407f": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_074a041d836046219e201ab8443e994d",
+            "placeholder": "​",
+            "style": "IPY_MODEL_3973b1ea9bdd42319bea13839b04ff73",
+            "value": "Generating train split: "
+          }
+        },
+        "dff1f82298b0499f92ac781678245018": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "FloatProgressModel",
+          "model_module_version": "1.5.0",
+          "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_ac0960ad1e8b42f29fe0342afe9a5dea",
+            "max": 1,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_36bd2c9c60884f6fac47d99ac8e62df7",
+            "value": 1
+          }
+        },
+        "8530897eee1f4512b95617d5e81c890d": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_35b591964bbd436da1eed4cb18063085",
+            "placeholder": "​",
+            "style": "IPY_MODEL_97e38497aedd4e52a2969a75f40e69cf",
+            "value": " 1095/0 [00:00&lt;00:00, 4172.58 examples/s]"
+          }
+        },
+        "266ee16a3cda44bfb634366770e52ca2": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "074a041d836046219e201ab8443e994d": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "3973b1ea9bdd42319bea13839b04ff73": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "ac0960ad1e8b42f29fe0342afe9a5dea": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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": "20px"
+          }
+        },
+        "36bd2c9c60884f6fac47d99ac8e62df7": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ProgressStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "35b591964bbd436da1eed4cb18063085": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "97e38497aedd4e52a2969a75f40e69cf": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "f63f4d33706a4a6bb275e895cce1399a": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HBoxModel",
+          "model_module_version": "1.5.0",
+          "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_cc16e3c53da9411aa5e705cb827a1ee0",
+              "IPY_MODEL_1b415516e0b74d70bed24553c562652e",
+              "IPY_MODEL_0e0f7993c16b44689184af9c92420ab7"
+            ],
+            "layout": "IPY_MODEL_073a3fe3ace242fb96da7338a8769316"
+          }
+        },
+        "cc16e3c53da9411aa5e705cb827a1ee0": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_d5d56db4f0394ec0baf682d3367d1750",
+            "placeholder": "​",
+            "style": "IPY_MODEL_9698cc34f7114de58518334c1e4acc79",
+            "value": "Generating validation split: "
+          }
+        },
+        "1b415516e0b74d70bed24553c562652e": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "FloatProgressModel",
+          "model_module_version": "1.5.0",
+          "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_dc8e7b2d8bbf44e5b8d94c9f4641928b",
+            "max": 1,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_41f87857e69641c7bd469f9305fbb77f",
+            "value": 1
+          }
+        },
+        "0e0f7993c16b44689184af9c92420ab7": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_ae4f97637ec84bf786780865cb7590aa",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5c4d39b44cce45debd1c79320f99265a",
+            "value": " 237/0 [00:00&lt;00:00, 1833.42 examples/s]"
+          }
+        },
+        "073a3fe3ace242fb96da7338a8769316": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "d5d56db4f0394ec0baf682d3367d1750": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "9698cc34f7114de58518334c1e4acc79": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "dc8e7b2d8bbf44e5b8d94c9f4641928b": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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": "20px"
+          }
+        },
+        "41f87857e69641c7bd469f9305fbb77f": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ProgressStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "ae4f97637ec84bf786780865cb7590aa": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "5c4d39b44cce45debd1c79320f99265a": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "918da148fd1d42cd80e890600d7a7de9": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HBoxModel",
+          "model_module_version": "1.5.0",
+          "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_81b7e458c813477fbf2994ae8a5202ba",
+              "IPY_MODEL_f1448623a7074e29af9b7bdcfd1ba52c",
+              "IPY_MODEL_100aa10071bf49098bc82ffcf34be553"
+            ],
+            "layout": "IPY_MODEL_6e9f3a291822403a8e25c243edca3d6a"
+          }
+        },
+        "81b7e458c813477fbf2994ae8a5202ba": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_4a69736428824330bc41a08fa8f42684",
+            "placeholder": "​",
+            "style": "IPY_MODEL_582d77ff31aa48d28913f8d19fa402a7",
+            "value": "Generating test split: "
+          }
+        },
+        "f1448623a7074e29af9b7bdcfd1ba52c": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "FloatProgressModel",
+          "model_module_version": "1.5.0",
+          "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_a30cc578ba9c48f4b046731fb47df0af",
+            "max": 1,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_5f8e24fef9fc459db426d955f002a764",
+            "value": 1
+          }
+        },
+        "100aa10071bf49098bc82ffcf34be553": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "HTMLModel",
+          "model_module_version": "1.5.0",
+          "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_2b04db4915fc4c7fa54a6ff0e5efed03",
+            "placeholder": "​",
+            "style": "IPY_MODEL_9c8a77cd6bce4d9a8e3c81714cad6964",
+            "value": " 244/0 [00:00&lt;00:00, 1721.26 examples/s]"
+          }
+        },
+        "6e9f3a291822403a8e25c243edca3d6a": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "4a69736428824330bc41a08fa8f42684": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "582d77ff31aa48d28913f8d19fa402a7": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "a30cc578ba9c48f4b046731fb47df0af": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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": "20px"
+          }
+        },
+        "5f8e24fef9fc459db426d955f002a764": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ProgressStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        },
+        "2b04db4915fc4c7fa54a6ff0e5efed03": {
+          "model_module": "@jupyter-widgets/base",
+          "model_name": "LayoutModel",
+          "model_module_version": "1.2.0",
+          "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
+          }
+        },
+        "9c8a77cd6bce4d9a8e3c81714cad6964": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "DescriptionStyleModel",
+          "model_module_version": "1.5.0",
+          "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": ""
+          }
+        }
+      }
+    }
+  },
+  "nbformat": 4,
+  "nbformat_minor": 0
+}
\ No newline at end of file

From 5c93b284545bab2f22c1a1821c96b023a9fbe2a7 Mon Sep 17 00:00:00 2001
From: boyu-cohere <boyu@cohere.com>
Date: Wed, 13 Mar 2024 18:16:52 -0400
Subject: [PATCH 3/3] fixed typo

---
 ...eeting_Summaries_General_&_LangChain.ipynb | 1564 ++++++++---------
 1 file changed, 781 insertions(+), 783 deletions(-)

diff --git a/notebooks/Meeting_Summaries_General_&_LangChain.ipynb b/notebooks/Meeting_Summaries_General_&_LangChain.ipynb
index e6d1d2e94..e3c071be5 100644
--- a/notebooks/Meeting_Summaries_General_&_LangChain.ipynb
+++ b/notebooks/Meeting_Summaries_General_&_LangChain.ipynb
@@ -198,8 +198,8 @@
       },
       "outputs": [
         {
-          "output_type": "stream",
           "name": "stderr",
+          "output_type": "stream",
           "text": [
             "/usr/local/lib/python3.10/dist-packages/huggingface_hub/utils/_token.py:88: UserWarning: \n",
             "The secret `HF_TOKEN` does not exist in your Colab secrets.\n",
@@ -210,106 +210,106 @@
           ]
         },
         {
-          "output_type": "display_data",
           "data": {
-            "text/plain": [
-              "Downloading readme:   0%|          | 0.00/21.0 [00:00<?, ?B/s]"
-            ],
             "application/vnd.jupyter.widget-view+json": {
+              "model_id": "43225bc344ed4374931511b8630baae4",
               "version_major": 2,
-              "version_minor": 0,
-              "model_id": "43225bc344ed4374931511b8630baae4"
-            }
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Downloading readme:   0%|          | 0.00/21.0 [00:00<?, ?B/s]"
+            ]
           },
-          "metadata": {}
+          "metadata": {},
+          "output_type": "display_data"
         },
         {
-          "output_type": "display_data",
           "data": {
-            "text/plain": [
-              "Downloading data:   0%|          | 0.00/4.45M [00:00<?, ?B/s]"
-            ],
             "application/vnd.jupyter.widget-view+json": {
+              "model_id": "23801508bc9c414cb106b6809c5f3043",
               "version_major": 2,
-              "version_minor": 0,
-              "model_id": "23801508bc9c414cb106b6809c5f3043"
-            }
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/4.45M [00:00<?, ?B/s]"
+            ]
           },
-          "metadata": {}
+          "metadata": {},
+          "output_type": "display_data"
         },
         {
-          "output_type": "display_data",
           "data": {
-            "text/plain": [
-              "Downloading data:   0%|          | 0.00/950k [00:00<?, ?B/s]"
-            ],
             "application/vnd.jupyter.widget-view+json": {
+              "model_id": "0d41cdf4e8b049d48912ee8ae20e5bf4",
               "version_major": 2,
-              "version_minor": 0,
-              "model_id": "0d41cdf4e8b049d48912ee8ae20e5bf4"
-            }
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/950k [00:00<?, ?B/s]"
+            ]
           },
-          "metadata": {}
+          "metadata": {},
+          "output_type": "display_data"
         },
         {
-          "output_type": "display_data",
           "data": {
-            "text/plain": [
-              "Downloading data:   0%|          | 0.00/917k [00:00<?, ?B/s]"
-            ],
             "application/vnd.jupyter.widget-view+json": {
+              "model_id": "fd57fbd3688942ae9a4a2c0c65817c88",
               "version_major": 2,
-              "version_minor": 0,
-              "model_id": "fd57fbd3688942ae9a4a2c0c65817c88"
-            }
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/917k [00:00<?, ?B/s]"
+            ]
           },
-          "metadata": {}
+          "metadata": {},
+          "output_type": "display_data"
         },
         {
-          "output_type": "display_data",
           "data": {
-            "text/plain": [
-              "Generating train split: 0 examples [00:00, ? examples/s]"
-            ],
             "application/vnd.jupyter.widget-view+json": {
+              "model_id": "6ed3be205c90420bbe4a2ff7911ef4ec",
               "version_major": 2,
-              "version_minor": 0,
-              "model_id": "6ed3be205c90420bbe4a2ff7911ef4ec"
-            }
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Generating train split: 0 examples [00:00, ? examples/s]"
+            ]
           },
-          "metadata": {}
+          "metadata": {},
+          "output_type": "display_data"
         },
         {
-          "output_type": "display_data",
           "data": {
-            "text/plain": [
-              "Generating validation split: 0 examples [00:00, ? examples/s]"
-            ],
             "application/vnd.jupyter.widget-view+json": {
+              "model_id": "f63f4d33706a4a6bb275e895cce1399a",
               "version_major": 2,
-              "version_minor": 0,
-              "model_id": "f63f4d33706a4a6bb275e895cce1399a"
-            }
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Generating validation split: 0 examples [00:00, ? examples/s]"
+            ]
           },
-          "metadata": {}
+          "metadata": {},
+          "output_type": "display_data"
         },
         {
-          "output_type": "display_data",
           "data": {
-            "text/plain": [
-              "Generating test split: 0 examples [00:00, ? examples/s]"
-            ],
             "application/vnd.jupyter.widget-view+json": {
+              "model_id": "918da148fd1d42cd80e890600d7a7de9",
               "version_major": 2,
-              "version_minor": 0,
-              "model_id": "918da148fd1d42cd80e890600d7a7de9"
-            }
+              "version_minor": 0
+            },
+            "text/plain": [
+              "Generating test split: 0 examples [00:00, ? examples/s]"
+            ]
           },
-          "metadata": {}
+          "metadata": {},
+          "output_type": "display_data"
         },
         {
-          "output_type": "stream",
           "name": "stdout",
+          "output_type": "stream",
           "text": [
             "PhD D: Sure because I I need a lot of time to to put the label or to do that\n",
             "Professor E: I mean we we know that there s noise There s there s continual noise from fans and so\n",
@@ -396,18 +396,12 @@
         "id": "TVWdxs3IIqpr"
       },
       "source": [
-        "If you've loaded the QMSum dataset, you'll see a back-and-forth dialogue between a Project Manager, an Industrial Designer, and someone responsible for the User Interface. They discuss the design of a banana-shaped remote controller.\n",
-        "\n",
         "Let's also check how many tokens that transcript corresponds to."
       ]
     },
     {
       "cell_type": "code",
-      "source": [
-        "# Measure the number of tokens\n",
-        "num_tokens = len(co.tokenize(transcript, model=co_model).tokens)\n",
-        "pprint(f\"Number of tokens: {num_tokens}\")"
-      ],
+      "execution_count": 5,
       "metadata": {
         "colab": {
           "base_uri": "https://localhost:8080/"
@@ -415,15 +409,19 @@
         "id": "eGlS8mMX5Kdw",
         "outputId": "844aa95f-8a81-48e0-c8fd-6dea58bd33c7"
       },
-      "execution_count": 5,
       "outputs": [
         {
-          "output_type": "stream",
           "name": "stdout",
+          "output_type": "stream",
           "text": [
             "Number of tokens: 1152\n"
           ]
         }
+      ],
+      "source": [
+        "# Measure the number of tokens\n",
+        "num_tokens = len(co.tokenize(transcript, model=co_model).tokens)\n",
+        "pprint(f\"Number of tokens: {num_tokens}\")"
       ]
     },
     {
@@ -441,16 +439,16 @@
       "cell_type": "code",
       "execution_count": 6,
       "metadata": {
-        "id": "QAFIK0OP8vwU",
         "colab": {
           "base_uri": "https://localhost:8080/"
         },
+        "id": "QAFIK0OP8vwU",
         "outputId": "37300916-0f32-4826-9bcf-608142ce0676"
       },
       "outputs": [
         {
-          "output_type": "stream",
           "name": "stdout",
+          "output_type": "stream",
           "text": [
             "The team discusses the time-consuming process of identifying instances of speaker overlap in audio\n",
             "recordings. PhD C suggests using close-talking microphones and automated methods to establish ground\n",
@@ -513,8 +511,8 @@
       },
       "outputs": [
         {
-          "output_type": "stream",
           "name": "stdout",
+          "output_type": "stream",
           "text": [
             "```json\n",
             "{\n",
@@ -553,14 +551,14 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "inBjpwOK6pTN"
+      },
       "source": [
         "That worked well! Asking for a structured response like a JSON object is a great approach for use cases where you will need to parse or post-process the output for use in downstream applications. To ensure the best possible completion, I provided a template of the expected JSON that the model filled out.\n",
         "\n",
         "Let's also try asking for a short 1 sentence summary."
-      ],
-      "metadata": {
-        "id": "inBjpwOK6pTN"
-      }
+      ]
     },
     {
       "cell_type": "code",
@@ -574,8 +572,8 @@
       },
       "outputs": [
         {
-          "output_type": "stream",
           "name": "stdout",
+          "output_type": "stream",
           "text": [
             "The team discusses using various methods, including close-talking mics and threshold-based volume\n",
             "filters, to accurately identify instances of speaker overlap in recorded meetings.\n"
@@ -599,16 +597,49 @@
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "KOaKiT6Ztkbz"
+      },
       "source": [
         "## LangChain summarization\n",
         "It's also very easy to use Command-R with LangChain for summarization."
-      ],
-      "metadata": {
-        "id": "KOaKiT6Ztkbz"
-      }
+      ]
     },
     {
       "cell_type": "code",
+      "execution_count": 9,
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "5eJM29wWtx-m",
+        "outputId": "3f29b633-e406-4608-95d3-60653ff41c51"
+      },
+      "outputs": [
+        {
+          "name": "stderr",
+          "output_type": "stream",
+          "text": [
+            "/usr/local/lib/python3.10/dist-packages/langchain_core/_api/deprecation.py:117: LangChainDeprecationWarning: The function `run` was deprecated in LangChain 0.1.0 and will be removed in 0.2.0. Use invoke instead.\n",
+            "  warn_deprecated(\n"
+          ]
+        },
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "\n",
+            "\n",
+            "\u001b[1m> Entering new StuffDocumentsChain chain...\u001b[0m\n",
+            "\n",
+            "\u001b[1m> Finished chain.\u001b[0m\n",
+            "The team discusses the time-consuming process of identifying instances of speaker overlap in audio\n",
+            "recordings. PhD C suggests using close-talking microphones and automated methods to establish ground\n",
+            "truth data, which Grad G is working on. This data would help develop a more efficient system than\n",
+            "manual marking.\n"
+          ]
+        }
+      ],
       "source": [
         "from langchain.chains.combine_documents.stuff import StuffDocumentsChain\n",
         "from langchain.chains.llm import LLMChain\n",
@@ -641,53 +672,23 @@
         "docs = [Document(page_content=transcript)]\n",
         "\n",
         "pprint(stuff_chain.run(docs))"
-      ],
-      "metadata": {
-        "id": "5eJM29wWtx-m",
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "outputId": "3f29b633-e406-4608-95d3-60653ff41c51"
-      },
-      "execution_count": 9,
-      "outputs": [
-        {
-          "output_type": "stream",
-          "name": "stderr",
-          "text": [
-            "/usr/local/lib/python3.10/dist-packages/langchain_core/_api/deprecation.py:117: LangChainDeprecationWarning: The function `run` was deprecated in LangChain 0.1.0 and will be removed in 0.2.0. Use invoke instead.\n",
-            "  warn_deprecated(\n"
-          ]
-        },
-        {
-          "output_type": "stream",
-          "name": "stdout",
-          "text": [
-            "\n",
-            "\n",
-            "\u001b[1m> Entering new StuffDocumentsChain chain...\u001b[0m\n",
-            "\n",
-            "\u001b[1m> Finished chain.\u001b[0m\n",
-            "The team discusses the time-consuming process of identifying instances of speaker overlap in audio\n",
-            "recordings. PhD C suggests using close-talking microphones and automated methods to establish ground\n",
-            "truth data, which Grad G is working on. This data would help develop a more efficient system than\n",
-            "manual marking.\n"
-          ]
-        }
       ]
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "MtS7tZi9t-dY"
+      },
       "source": [
         "Command-R has a context window of 128K tokens, so you can use very long documents, or multiple documents, all in a single API call.\n",
         "\n"
-      ],
-      "metadata": {
-        "id": "MtS7tZi9t-dY"
-      }
+      ]
     },
     {
       "cell_type": "markdown",
+      "metadata": {
+        "id": "uj8kdy5auKKY"
+      },
       "source": [
         "## Wrap up\n",
         "\n",
@@ -698,10 +699,7 @@
         "In cases where you can't fit the input text into the context window, see our [cookbook for long document strategies](https://colab.research.google.com/drive/1zxSAbruOWwWJHNsj3N56uxZtUeiS7Evd#scrollTo=4JgYYTg7Shvq).\n",
         "\n",
         "For more information, you can reach out to us at summarize@cohere.com!"
-      ],
-      "metadata": {
-        "id": "uj8kdy5auKKY"
-      }
+      ]
     }
   ],
   "metadata": {
@@ -726,98 +724,10 @@
     },
     "widgets": {
       "application/vnd.jupyter.widget-state+json": {
-        "43225bc344ed4374931511b8630baae4": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HBoxModel",
-          "model_module_version": "1.5.0",
-          "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_56d2b7a171c4497a83e26bbafbbb4612",
-              "IPY_MODEL_2f73ed56c08a45c9ab1dbdfb59eb51cd",
-              "IPY_MODEL_4798a0f8e4d24a568fded08cde65dc7a"
-            ],
-            "layout": "IPY_MODEL_68aa82f9c86f47b6a810aa0404bb8266"
-          }
-        },
-        "56d2b7a171c4497a83e26bbafbbb4612": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
-          "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_09fa50ba56b6483d88413799d45e6baa",
-            "placeholder": "​",
-            "style": "IPY_MODEL_079f8238a7124707963f240aaef7eb4f",
-            "value": "Downloading readme: 100%"
-          }
-        },
-        "2f73ed56c08a45c9ab1dbdfb59eb51cd": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "FloatProgressModel",
-          "model_module_version": "1.5.0",
-          "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_904ddfbc8af44e43bb32a7c14639de8d",
-            "max": 21,
-            "min": 0,
-            "orientation": "horizontal",
-            "style": "IPY_MODEL_38a4dabfd7b648dca78f2d6159079779",
-            "value": 21
-          }
-        },
-        "4798a0f8e4d24a568fded08cde65dc7a": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
-          "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_7922f387c6a04bfc97bdd046097b567d",
-            "placeholder": "​",
-            "style": "IPY_MODEL_95d74166787244d4807e5c4b0223ab4c",
-            "value": " 21.0/21.0 [00:00&lt;00:00, 767B/s]"
-          }
-        },
-        "68aa82f9c86f47b6a810aa0404bb8266": {
+        "073a3fe3ace242fb96da7338a8769316": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -866,10 +776,10 @@
             "width": null
           }
         },
-        "09fa50ba56b6483d88413799d45e6baa": {
+        "074a041d836046219e201ab8443e994d": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -920,8 +830,8 @@
         },
         "079f8238a7124707963f240aaef7eb4f": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "DescriptionStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
@@ -933,10 +843,10 @@
             "description_width": ""
           }
         },
-        "904ddfbc8af44e43bb32a7c14639de8d": {
+        "09fa50ba56b6483d88413799d45e6baa": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -985,26 +895,10 @@
             "width": null
           }
         },
-        "38a4dabfd7b648dca78f2d6159079779": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "ProgressStyleModel",
-          "model_module_version": "1.5.0",
-          "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": ""
-          }
-        },
-        "7922f387c6a04bfc97bdd046097b567d": {
+        "0cdd8ccc3bef493ba59b3ca0cae5f785": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1053,10 +947,10 @@
             "width": null
           }
         },
-        "95d74166787244d4807e5c4b0223ab4c": {
+        "0ce661f36e724d24a65d875106b3e92d": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "DescriptionStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
@@ -1068,10 +962,10 @@
             "description_width": ""
           }
         },
-        "23801508bc9c414cb106b6809c5f3043": {
+        "0d41cdf4e8b049d48912ee8ae20e5bf4": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "HBoxModel",
           "model_module_version": "1.5.0",
+          "model_name": "HBoxModel",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -1083,17 +977,38 @@
             "_view_name": "HBoxView",
             "box_style": "",
             "children": [
-              "IPY_MODEL_2e1ca43dfaf14957ad788519f57333e9",
-              "IPY_MODEL_9df7c90e2a9444c1b97dc055bb805a4e",
-              "IPY_MODEL_5f9d1e97dd62416587abfc724743c685"
+              "IPY_MODEL_dc022d95fff345e88fe3fdeb42ad78c8",
+              "IPY_MODEL_73fa502143aa4186a32bcf58fdfe0424",
+              "IPY_MODEL_88abb7063a4f4f43b48e51ba746ec6e6"
             ],
-            "layout": "IPY_MODEL_6b68bf1fc952421c8534b262dc6b27b9"
+            "layout": "IPY_MODEL_a2f91869ede34418ad6d934680a4a447"
           }
         },
-        "2e1ca43dfaf14957ad788519f57333e9": {
+        "0e0f7993c16b44689184af9c92420ab7": {
           "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_ae4f97637ec84bf786780865cb7590aa",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5c4d39b44cce45debd1c79320f99265a",
+            "value": " 237/0 [00:00&lt;00:00, 1833.42 examples/s]"
+          }
+        },
+        "100aa10071bf49098bc82ffcf34be553": {
+          "model_module": "@jupyter-widgets/controls",
           "model_module_version": "1.5.0",
+          "model_name": "HTMLModel",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -1105,16 +1020,16 @@
             "_view_name": "HTMLView",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_5f2d4056e0be4205bd09948715a66f07",
+            "layout": "IPY_MODEL_2b04db4915fc4c7fa54a6ff0e5efed03",
             "placeholder": "​",
-            "style": "IPY_MODEL_5f3a09f953e64b44bb91a8258d91ef28",
-            "value": "Downloading data: 100%"
+            "style": "IPY_MODEL_9c8a77cd6bce4d9a8e3c81714cad6964",
+            "value": " 244/0 [00:00&lt;00:00, 1721.26 examples/s]"
           }
         },
-        "9df7c90e2a9444c1b97dc055bb805a4e": {
+        "1b415516e0b74d70bed24553c562652e": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "FloatProgressModel",
           "model_module_version": "1.5.0",
+          "model_name": "FloatProgressModel",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -1127,39 +1042,40 @@
             "bar_style": "success",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_5521d0c2117a4500b741197c99da7d36",
-            "max": 4445270,
+            "layout": "IPY_MODEL_dc8e7b2d8bbf44e5b8d94c9f4641928b",
+            "max": 1,
             "min": 0,
             "orientation": "horizontal",
-            "style": "IPY_MODEL_99872e6b03a047a38b1ea88976c4b1a4",
-            "value": 4445270
+            "style": "IPY_MODEL_41f87857e69641c7bd469f9305fbb77f",
+            "value": 1
           }
         },
-        "5f9d1e97dd62416587abfc724743c685": {
+        "23801508bc9c414cb106b6809c5f3043": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
           "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": "HTMLModel",
+            "_model_name": "HBoxModel",
             "_view_count": null,
             "_view_module": "@jupyter-widgets/controls",
             "_view_module_version": "1.5.0",
-            "_view_name": "HTMLView",
-            "description": "",
-            "description_tooltip": null,
-            "layout": "IPY_MODEL_a4bc752fd44a4a0da969d08fd232644a",
-            "placeholder": "​",
-            "style": "IPY_MODEL_cc7e4555f04f4d09aab780af5dce5004",
-            "value": " 4.45M/4.45M [00:00&lt;00:00, 5.31MB/s]"
+            "_view_name": "HBoxView",
+            "box_style": "",
+            "children": [
+              "IPY_MODEL_2e1ca43dfaf14957ad788519f57333e9",
+              "IPY_MODEL_9df7c90e2a9444c1b97dc055bb805a4e",
+              "IPY_MODEL_5f9d1e97dd62416587abfc724743c685"
+            ],
+            "layout": "IPY_MODEL_6b68bf1fc952421c8534b262dc6b27b9"
           }
         },
-        "6b68bf1fc952421c8534b262dc6b27b9": {
+        "266ee16a3cda44bfb634366770e52ca2": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1208,10 +1124,31 @@
             "width": null
           }
         },
-        "5f2d4056e0be4205bd09948715a66f07": {
+        "28df182d45c54455974d58435ea5407f": {
+          "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_074a041d836046219e201ab8443e994d",
+            "placeholder": "​",
+            "style": "IPY_MODEL_3973b1ea9bdd42319bea13839b04ff73",
+            "value": "Generating train split: "
+          }
+        },
+        "2b04db4915fc4c7fa54a6ff0e5efed03": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1260,25 +1197,76 @@
             "width": null
           }
         },
-        "5f3a09f953e64b44bb91a8258d91ef28": {
+        "2e1ca43dfaf14957ad788519f57333e9": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "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": "DescriptionStyleModel",
+            "_model_name": "HTMLModel",
             "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "description_width": ""
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "HTMLView",
+            "description": "",
+            "description_tooltip": null,
+            "layout": "IPY_MODEL_5f2d4056e0be4205bd09948715a66f07",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5f3a09f953e64b44bb91a8258d91ef28",
+            "value": "Downloading data: 100%"
           }
         },
-        "5521d0c2117a4500b741197c99da7d36": {
+        "2e9d71ebf60049a68dce99ae57633ed6": {
+          "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_b9284cee75b24777815d37512975fa64",
+            "placeholder": "​",
+            "style": "IPY_MODEL_e8e40b743abc4c0d8f0fcc66a2d45c30",
+            "value": " 917k/917k [00:00&lt;00:00, 1.64MB/s]"
+          }
+        },
+        "2f73ed56c08a45c9ab1dbdfb59eb51cd": {
+          "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_904ddfbc8af44e43bb32a7c14639de8d",
+            "max": 21,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_38a4dabfd7b648dca78f2d6159079779",
+            "value": 21
+          }
+        },
+        "35b591964bbd436da1eed4cb18063085": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1327,10 +1315,26 @@
             "width": null
           }
         },
-        "99872e6b03a047a38b1ea88976c4b1a4": {
+        "36bd2c9c60884f6fac47d99ac8e62df7": {
           "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": ""
+          }
+        },
+        "38a4dabfd7b648dca78f2d6159079779": {
+          "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",
@@ -1343,10 +1347,25 @@
             "description_width": ""
           }
         },
-        "a4bc752fd44a4a0da969d08fd232644a": {
+        "3973b1ea9bdd42319bea13839b04ff73": {
+          "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": ""
+          }
+        },
+        "3eaf53e3d6674527aaca6f4aeda71f7d": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1395,25 +1414,26 @@
             "width": null
           }
         },
-        "cc7e4555f04f4d09aab780af5dce5004": {
+        "41f87857e69641c7bd469f9305fbb77f": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "ProgressStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
-            "_model_name": "DescriptionStyleModel",
+            "_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": ""
           }
         },
-        "0d41cdf4e8b049d48912ee8ae20e5bf4": {
+        "43225bc344ed4374931511b8630baae4": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "HBoxModel",
           "model_module_version": "1.5.0",
+          "model_name": "HBoxModel",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -1425,62 +1445,17 @@
             "_view_name": "HBoxView",
             "box_style": "",
             "children": [
-              "IPY_MODEL_dc022d95fff345e88fe3fdeb42ad78c8",
-              "IPY_MODEL_73fa502143aa4186a32bcf58fdfe0424",
-              "IPY_MODEL_88abb7063a4f4f43b48e51ba746ec6e6"
+              "IPY_MODEL_56d2b7a171c4497a83e26bbafbbb4612",
+              "IPY_MODEL_2f73ed56c08a45c9ab1dbdfb59eb51cd",
+              "IPY_MODEL_4798a0f8e4d24a568fded08cde65dc7a"
             ],
-            "layout": "IPY_MODEL_a2f91869ede34418ad6d934680a4a447"
-          }
-        },
-        "dc022d95fff345e88fe3fdeb42ad78c8": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
-          "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_f1a3b77843af4b219e24f94fe7224516",
-            "placeholder": "​",
-            "style": "IPY_MODEL_5895c33db870460681eb05f98da67a5b",
-            "value": "Downloading data: 100%"
+            "layout": "IPY_MODEL_68aa82f9c86f47b6a810aa0404bb8266"
           }
         },
-        "73fa502143aa4186a32bcf58fdfe0424": {
+        "4798a0f8e4d24a568fded08cde65dc7a": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "FloatProgressModel",
           "model_module_version": "1.5.0",
-          "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_8ebf85a7dd4c4a538405100fb08c7376",
-            "max": 950100,
-            "min": 0,
-            "orientation": "horizontal",
-            "style": "IPY_MODEL_52b2f525cfdb429aa79cd7097989b657",
-            "value": 950100
-          }
-        },
-        "88abb7063a4f4f43b48e51ba746ec6e6": {
-          "model_module": "@jupyter-widgets/controls",
           "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -1492,16 +1467,16 @@
             "_view_name": "HTMLView",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_fef8c84b911a419c882ae7c0d8b6a937",
+            "layout": "IPY_MODEL_7922f387c6a04bfc97bdd046097b567d",
             "placeholder": "​",
-            "style": "IPY_MODEL_9cecfc3ecc2c4027a76182fb4330ea6a",
-            "value": " 950k/950k [00:00&lt;00:00, 1.80MB/s]"
+            "style": "IPY_MODEL_95d74166787244d4807e5c4b0223ab4c",
+            "value": " 21.0/21.0 [00:00&lt;00:00, 767B/s]"
           }
         },
-        "a2f91869ede34418ad6d934680a4a447": {
+        "4a69736428824330bc41a08fa8f42684": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1550,10 +1525,26 @@
             "width": null
           }
         },
-        "f1a3b77843af4b219e24f94fe7224516": {
+        "52b2f525cfdb429aa79cd7097989b657": {
+          "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": ""
+          }
+        },
+        "5521d0c2117a4500b741197c99da7d36": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1602,10 +1593,31 @@
             "width": null
           }
         },
-        "5895c33db870460681eb05f98da67a5b": {
+        "56d2b7a171c4497a83e26bbafbbb4612": {
+          "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_09fa50ba56b6483d88413799d45e6baa",
+            "placeholder": "​",
+            "style": "IPY_MODEL_079f8238a7124707963f240aaef7eb4f",
+            "value": "Downloading readme: 100%"
+          }
+        },
+        "582d77ff31aa48d28913f8d19fa402a7": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "DescriptionStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
@@ -1617,78 +1629,40 @@
             "description_width": ""
           }
         },
-        "8ebf85a7dd4c4a538405100fb08c7376": {
-          "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
-          "model_module_version": "1.2.0",
+        "5895c33db870460681eb05f98da67a5b": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_module_version": "1.5.0",
+          "model_name": "DescriptionStyleModel",
           "state": {
-            "_model_module": "@jupyter-widgets/base",
-            "_model_module_version": "1.2.0",
-            "_model_name": "LayoutModel",
+            "_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": "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
+            "_view_name": "StyleView",
+            "description_width": ""
           }
         },
-        "52b2f525cfdb429aa79cd7097989b657": {
+        "5c4d39b44cce45debd1c79320f99265a": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "ProgressStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "DescriptionStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
-            "_model_name": "ProgressStyleModel",
+            "_model_name": "DescriptionStyleModel",
             "_view_count": null,
             "_view_module": "@jupyter-widgets/base",
             "_view_module_version": "1.2.0",
             "_view_name": "StyleView",
-            "bar_color": null,
             "description_width": ""
           }
         },
-        "fef8c84b911a419c882ae7c0d8b6a937": {
+        "5f2d4056e0be4205bd09948715a66f07": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1737,10 +1711,10 @@
             "width": null
           }
         },
-        "9cecfc3ecc2c4027a76182fb4330ea6a": {
+        "5f3a09f953e64b44bb91a8258d91ef28": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "DescriptionStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
@@ -1752,32 +1726,26 @@
             "description_width": ""
           }
         },
-        "fd57fbd3688942ae9a4a2c0c65817c88": {
+        "5f8e24fef9fc459db426d955f002a764": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "HBoxModel",
           "model_module_version": "1.5.0",
+          "model_name": "ProgressStyleModel",
           "state": {
-            "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
-            "_model_name": "HBoxModel",
+            "_model_name": "ProgressStyleModel",
             "_view_count": null,
-            "_view_module": "@jupyter-widgets/controls",
-            "_view_module_version": "1.5.0",
-            "_view_name": "HBoxView",
-            "box_style": "",
-            "children": [
-              "IPY_MODEL_af362384efb04c50bd2e3d16abb28448",
-              "IPY_MODEL_671324a8ab0249a59e9d10f7b7df0d3a",
-              "IPY_MODEL_2e9d71ebf60049a68dce99ae57633ed6"
-            ],
-            "layout": "IPY_MODEL_3eaf53e3d6674527aaca6f4aeda71f7d"
+            "_view_module": "@jupyter-widgets/base",
+            "_view_module_version": "1.2.0",
+            "_view_name": "StyleView",
+            "bar_color": null,
+            "description_width": ""
           }
         },
-        "af362384efb04c50bd2e3d16abb28448": {
+        "5f9d1e97dd62416587abfc724743c685": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
           "model_module_version": "1.5.0",
+          "model_name": "HTMLModel",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -1789,16 +1757,16 @@
             "_view_name": "HTMLView",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_e829f283ba864af1af50cdce847692c1",
+            "layout": "IPY_MODEL_a4bc752fd44a4a0da969d08fd232644a",
             "placeholder": "​",
-            "style": "IPY_MODEL_0ce661f36e724d24a65d875106b3e92d",
-            "value": "Downloading data: 100%"
+            "style": "IPY_MODEL_cc7e4555f04f4d09aab780af5dce5004",
+            "value": " 4.45M/4.45M [00:00&lt;00:00, 5.31MB/s]"
           }
         },
         "671324a8ab0249a59e9d10f7b7df0d3a": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "FloatProgressModel",
           "model_module_version": "1.5.0",
+          "model_name": "FloatProgressModel",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -1819,31 +1787,10 @@
             "value": 916846
           }
         },
-        "2e9d71ebf60049a68dce99ae57633ed6": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
-          "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_b9284cee75b24777815d37512975fa64",
-            "placeholder": "​",
-            "style": "IPY_MODEL_e8e40b743abc4c0d8f0fcc66a2d45c30",
-            "value": " 917k/917k [00:00&lt;00:00, 1.64MB/s]"
-          }
-        },
-        "3eaf53e3d6674527aaca6f4aeda71f7d": {
+        "68aa82f9c86f47b6a810aa0404bb8266": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1892,10 +1839,10 @@
             "width": null
           }
         },
-        "e829f283ba864af1af50cdce847692c1": {
+        "6b68bf1fc952421c8534b262dc6b27b9": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -1944,25 +1891,10 @@
             "width": null
           }
         },
-        "0ce661f36e724d24a65d875106b3e92d": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
-          "model_module_version": "1.5.0",
-          "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": ""
-          }
-        },
-        "0cdd8ccc3bef493ba59b3ca0cae5f785": {
+        "6e9f3a291822403a8e25c243edca3d6a": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2011,10 +1943,32 @@
             "width": null
           }
         },
+        "6ed3be205c90420bbe4a2ff7911ef4ec": {
+          "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_28df182d45c54455974d58435ea5407f",
+              "IPY_MODEL_dff1f82298b0499f92ac781678245018",
+              "IPY_MODEL_8530897eee1f4512b95617d5e81c890d"
+            ],
+            "layout": "IPY_MODEL_266ee16a3cda44bfb634366770e52ca2"
+          }
+        },
         "6ed77172d5da420093304ccde9420f98": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "ProgressStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "ProgressStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
@@ -2027,10 +1981,34 @@
             "description_width": ""
           }
         },
-        "b9284cee75b24777815d37512975fa64": {
+        "73fa502143aa4186a32bcf58fdfe0424": {
+          "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_8ebf85a7dd4c4a538405100fb08c7376",
+            "max": 950100,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_52b2f525cfdb429aa79cd7097989b657",
+            "value": 950100
+          }
+        },
+        "7922f387c6a04bfc97bdd046097b567d": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2079,47 +2057,10 @@
             "width": null
           }
         },
-        "e8e40b743abc4c0d8f0fcc66a2d45c30": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
-          "model_module_version": "1.5.0",
-          "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": ""
-          }
-        },
-        "6ed3be205c90420bbe4a2ff7911ef4ec": {
+        "81b7e458c813477fbf2994ae8a5202ba": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "HBoxModel",
           "model_module_version": "1.5.0",
-          "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_28df182d45c54455974d58435ea5407f",
-              "IPY_MODEL_dff1f82298b0499f92ac781678245018",
-              "IPY_MODEL_8530897eee1f4512b95617d5e81c890d"
-            ],
-            "layout": "IPY_MODEL_266ee16a3cda44bfb634366770e52ca2"
-          }
-        },
-        "28df182d45c54455974d58435ea5407f": {
-          "model_module": "@jupyter-widgets/controls",
           "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -2131,40 +2072,37 @@
             "_view_name": "HTMLView",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_074a041d836046219e201ab8443e994d",
+            "layout": "IPY_MODEL_4a69736428824330bc41a08fa8f42684",
             "placeholder": "​",
-            "style": "IPY_MODEL_3973b1ea9bdd42319bea13839b04ff73",
-            "value": "Generating train split: "
+            "style": "IPY_MODEL_582d77ff31aa48d28913f8d19fa402a7",
+            "value": "Generating test split: "
           }
         },
-        "dff1f82298b0499f92ac781678245018": {
+        "8530897eee1f4512b95617d5e81c890d": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "FloatProgressModel",
           "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": "FloatProgressModel",
+            "_model_name": "HTMLModel",
             "_view_count": null,
             "_view_module": "@jupyter-widgets/controls",
             "_view_module_version": "1.5.0",
-            "_view_name": "ProgressView",
-            "bar_style": "success",
+            "_view_name": "HTMLView",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_ac0960ad1e8b42f29fe0342afe9a5dea",
-            "max": 1,
-            "min": 0,
-            "orientation": "horizontal",
-            "style": "IPY_MODEL_36bd2c9c60884f6fac47d99ac8e62df7",
-            "value": 1
+            "layout": "IPY_MODEL_35b591964bbd436da1eed4cb18063085",
+            "placeholder": "​",
+            "style": "IPY_MODEL_97e38497aedd4e52a2969a75f40e69cf",
+            "value": " 1095/0 [00:00&lt;00:00, 4172.58 examples/s]"
           }
         },
-        "8530897eee1f4512b95617d5e81c890d": {
+        "88abb7063a4f4f43b48e51ba746ec6e6": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
           "model_module_version": "1.5.0",
+          "model_name": "HTMLModel",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -2176,16 +2114,16 @@
             "_view_name": "HTMLView",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_35b591964bbd436da1eed4cb18063085",
+            "layout": "IPY_MODEL_fef8c84b911a419c882ae7c0d8b6a937",
             "placeholder": "​",
-            "style": "IPY_MODEL_97e38497aedd4e52a2969a75f40e69cf",
-            "value": " 1095/0 [00:00&lt;00:00, 4172.58 examples/s]"
+            "style": "IPY_MODEL_9cecfc3ecc2c4027a76182fb4330ea6a",
+            "value": " 950k/950k [00:00&lt;00:00, 1.80MB/s]"
           }
         },
-        "266ee16a3cda44bfb634366770e52ca2": {
+        "8ebf85a7dd4c4a538405100fb08c7376": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2234,10 +2172,10 @@
             "width": null
           }
         },
-        "074a041d836046219e201ab8443e994d": {
+        "904ddfbc8af44e43bb32a7c14639de8d": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2286,10 +2224,47 @@
             "width": null
           }
         },
-        "3973b1ea9bdd42319bea13839b04ff73": {
+        "918da148fd1d42cd80e890600d7a7de9": {
+          "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_81b7e458c813477fbf2994ae8a5202ba",
+              "IPY_MODEL_f1448623a7074e29af9b7bdcfd1ba52c",
+              "IPY_MODEL_100aa10071bf49098bc82ffcf34be553"
+            ],
+            "layout": "IPY_MODEL_6e9f3a291822403a8e25c243edca3d6a"
+          }
+        },
+        "95d74166787244d4807e5c4b0223ab4c": {
           "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": ""
+          }
+        },
+        "9698cc34f7114de58518334c1e4acc79": {
+          "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",
@@ -2301,10 +2276,95 @@
             "description_width": ""
           }
         },
-        "ac0960ad1e8b42f29fe0342afe9a5dea": {
+        "97e38497aedd4e52a2969a75f40e69cf": {
+          "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": ""
+          }
+        },
+        "99872e6b03a047a38b1ea88976c4b1a4": {
+          "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": ""
+          }
+        },
+        "9c8a77cd6bce4d9a8e3c81714cad6964": {
+          "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": ""
+          }
+        },
+        "9cecfc3ecc2c4027a76182fb4330ea6a": {
+          "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": ""
+          }
+        },
+        "9df7c90e2a9444c1b97dc055bb805a4e": {
+          "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_5521d0c2117a4500b741197c99da7d36",
+            "max": 4445270,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_99872e6b03a047a38b1ea88976c4b1a4",
+            "value": 4445270
+          }
+        },
+        "a2f91869ede34418ad6d934680a4a447": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2350,29 +2410,65 @@
             "right": null,
             "top": null,
             "visibility": null,
-            "width": "20px"
+            "width": null
           }
         },
-        "36bd2c9c60884f6fac47d99ac8e62df7": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "ProgressStyleModel",
-          "model_module_version": "1.5.0",
+        "a30cc578ba9c48f4b046731fb47df0af": {
+          "model_module": "@jupyter-widgets/base",
+          "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
-            "_model_module": "@jupyter-widgets/controls",
-            "_model_module_version": "1.5.0",
-            "_model_name": "ProgressStyleModel",
+            "_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": "StyleView",
-            "bar_color": null,
-            "description_width": ""
+            "_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": "20px"
           }
         },
-        "35b591964bbd436da1eed4cb18063085": {
+        "a4bc752fd44a4a0da969d08fd232644a": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2421,113 +2517,10 @@
             "width": null
           }
         },
-        "97e38497aedd4e52a2969a75f40e69cf": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
-          "model_module_version": "1.5.0",
-          "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": ""
-          }
-        },
-        "f63f4d33706a4a6bb275e895cce1399a": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HBoxModel",
-          "model_module_version": "1.5.0",
-          "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_cc16e3c53da9411aa5e705cb827a1ee0",
-              "IPY_MODEL_1b415516e0b74d70bed24553c562652e",
-              "IPY_MODEL_0e0f7993c16b44689184af9c92420ab7"
-            ],
-            "layout": "IPY_MODEL_073a3fe3ace242fb96da7338a8769316"
-          }
-        },
-        "cc16e3c53da9411aa5e705cb827a1ee0": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
-          "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_d5d56db4f0394ec0baf682d3367d1750",
-            "placeholder": "​",
-            "style": "IPY_MODEL_9698cc34f7114de58518334c1e4acc79",
-            "value": "Generating validation split: "
-          }
-        },
-        "1b415516e0b74d70bed24553c562652e": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "FloatProgressModel",
-          "model_module_version": "1.5.0",
-          "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_dc8e7b2d8bbf44e5b8d94c9f4641928b",
-            "max": 1,
-            "min": 0,
-            "orientation": "horizontal",
-            "style": "IPY_MODEL_41f87857e69641c7bd469f9305fbb77f",
-            "value": 1
-          }
-        },
-        "0e0f7993c16b44689184af9c92420ab7": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
-          "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_ae4f97637ec84bf786780865cb7590aa",
-            "placeholder": "​",
-            "style": "IPY_MODEL_5c4d39b44cce45debd1c79320f99265a",
-            "value": " 237/0 [00:00&lt;00:00, 1833.42 examples/s]"
-          }
-        },
-        "073a3fe3ace242fb96da7338a8769316": {
+        "ac0960ad1e8b42f29fe0342afe9a5dea": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2573,13 +2566,13 @@
             "right": null,
             "top": null,
             "visibility": null,
-            "width": null
+            "width": "20px"
           }
         },
-        "d5d56db4f0394ec0baf682d3367d1750": {
+        "ae4f97637ec84bf786780865cb7590aa": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2628,25 +2621,31 @@
             "width": null
           }
         },
-        "9698cc34f7114de58518334c1e4acc79": {
+        "af362384efb04c50bd2e3d16abb28448": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "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": "DescriptionStyleModel",
+            "_model_name": "HTMLModel",
             "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "description_width": ""
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "HTMLView",
+            "description": "",
+            "description_tooltip": null,
+            "layout": "IPY_MODEL_e829f283ba864af1af50cdce847692c1",
+            "placeholder": "​",
+            "style": "IPY_MODEL_0ce661f36e724d24a65d875106b3e92d",
+            "value": "Downloading data: 100%"
           }
         },
-        "dc8e7b2d8bbf44e5b8d94c9f4641928b": {
+        "b9284cee75b24777815d37512975fa64": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2692,29 +2691,49 @@
             "right": null,
             "top": null,
             "visibility": null,
-            "width": "20px"
+            "width": null
           }
         },
-        "41f87857e69641c7bd469f9305fbb77f": {
+        "cc16e3c53da9411aa5e705cb827a1ee0": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "ProgressStyleModel",
           "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": "ProgressStyleModel",
+            "_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_d5d56db4f0394ec0baf682d3367d1750",
+            "placeholder": "​",
+            "style": "IPY_MODEL_9698cc34f7114de58518334c1e4acc79",
+            "value": "Generating validation split: "
+          }
+        },
+        "cc7e4555f04f4d09aab780af5dce5004": {
+          "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",
-            "bar_color": null,
             "description_width": ""
           }
         },
-        "ae4f97637ec84bf786780865cb7590aa": {
+        "d5d56db4f0394ec0baf682d3367d1750": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2763,92 +2782,10 @@
             "width": null
           }
         },
-        "5c4d39b44cce45debd1c79320f99265a": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
-          "model_module_version": "1.5.0",
-          "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": ""
-          }
-        },
-        "918da148fd1d42cd80e890600d7a7de9": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HBoxModel",
-          "model_module_version": "1.5.0",
-          "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_81b7e458c813477fbf2994ae8a5202ba",
-              "IPY_MODEL_f1448623a7074e29af9b7bdcfd1ba52c",
-              "IPY_MODEL_100aa10071bf49098bc82ffcf34be553"
-            ],
-            "layout": "IPY_MODEL_6e9f3a291822403a8e25c243edca3d6a"
-          }
-        },
-        "81b7e458c813477fbf2994ae8a5202ba": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
-          "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_4a69736428824330bc41a08fa8f42684",
-            "placeholder": "​",
-            "style": "IPY_MODEL_582d77ff31aa48d28913f8d19fa402a7",
-            "value": "Generating test split: "
-          }
-        },
-        "f1448623a7074e29af9b7bdcfd1ba52c": {
+        "dc022d95fff345e88fe3fdeb42ad78c8": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "FloatProgressModel",
           "model_module_version": "1.5.0",
-          "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_a30cc578ba9c48f4b046731fb47df0af",
-            "max": 1,
-            "min": 0,
-            "orientation": "horizontal",
-            "style": "IPY_MODEL_5f8e24fef9fc459db426d955f002a764",
-            "value": 1
-          }
-        },
-        "100aa10071bf49098bc82ffcf34be553": {
-          "model_module": "@jupyter-widgets/controls",
           "model_name": "HTMLModel",
-          "model_module_version": "1.5.0",
           "state": {
             "_dom_classes": [],
             "_model_module": "@jupyter-widgets/controls",
@@ -2860,16 +2797,16 @@
             "_view_name": "HTMLView",
             "description": "",
             "description_tooltip": null,
-            "layout": "IPY_MODEL_2b04db4915fc4c7fa54a6ff0e5efed03",
+            "layout": "IPY_MODEL_f1a3b77843af4b219e24f94fe7224516",
             "placeholder": "​",
-            "style": "IPY_MODEL_9c8a77cd6bce4d9a8e3c81714cad6964",
-            "value": " 244/0 [00:00&lt;00:00, 1721.26 examples/s]"
+            "style": "IPY_MODEL_5895c33db870460681eb05f98da67a5b",
+            "value": "Downloading data: 100%"
           }
         },
-        "6e9f3a291822403a8e25c243edca3d6a": {
+        "dc8e7b2d8bbf44e5b8d94c9f4641928b": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2915,13 +2852,37 @@
             "right": null,
             "top": null,
             "visibility": null,
-            "width": null
+            "width": "20px"
           }
         },
-        "4a69736428824330bc41a08fa8f42684": {
+        "dff1f82298b0499f92ac781678245018": {
+          "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_ac0960ad1e8b42f29fe0342afe9a5dea",
+            "max": 1,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_36bd2c9c60884f6fac47d99ac8e62df7",
+            "value": 1
+          }
+        },
+        "e829f283ba864af1af50cdce847692c1": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -2970,10 +2931,10 @@
             "width": null
           }
         },
-        "582d77ff31aa48d28913f8d19fa402a7": {
+        "e8e40b743abc4c0d8f0fcc66a2d45c30": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
           "model_module_version": "1.5.0",
+          "model_name": "DescriptionStyleModel",
           "state": {
             "_model_module": "@jupyter-widgets/controls",
             "_model_module_version": "1.5.0",
@@ -2985,10 +2946,34 @@
             "description_width": ""
           }
         },
-        "a30cc578ba9c48f4b046731fb47df0af": {
+        "f1448623a7074e29af9b7bdcfd1ba52c": {
+          "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_a30cc578ba9c48f4b046731fb47df0af",
+            "max": 1,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_5f8e24fef9fc459db426d955f002a764",
+            "value": 1
+          }
+        },
+        "f1a3b77843af4b219e24f94fe7224516": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -3034,29 +3019,57 @@
             "right": null,
             "top": null,
             "visibility": null,
-            "width": "20px"
+            "width": null
           }
         },
-        "5f8e24fef9fc459db426d955f002a764": {
+        "f63f4d33706a4a6bb275e895cce1399a": {
           "model_module": "@jupyter-widgets/controls",
-          "model_name": "ProgressStyleModel",
           "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": "ProgressStyleModel",
+            "_model_name": "HBoxModel",
             "_view_count": null,
-            "_view_module": "@jupyter-widgets/base",
-            "_view_module_version": "1.2.0",
-            "_view_name": "StyleView",
-            "bar_color": null,
-            "description_width": ""
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "HBoxView",
+            "box_style": "",
+            "children": [
+              "IPY_MODEL_cc16e3c53da9411aa5e705cb827a1ee0",
+              "IPY_MODEL_1b415516e0b74d70bed24553c562652e",
+              "IPY_MODEL_0e0f7993c16b44689184af9c92420ab7"
+            ],
+            "layout": "IPY_MODEL_073a3fe3ace242fb96da7338a8769316"
           }
         },
-        "2b04db4915fc4c7fa54a6ff0e5efed03": {
+        "fd57fbd3688942ae9a4a2c0c65817c88": {
+          "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_af362384efb04c50bd2e3d16abb28448",
+              "IPY_MODEL_671324a8ab0249a59e9d10f7b7df0d3a",
+              "IPY_MODEL_2e9d71ebf60049a68dce99ae57633ed6"
+            ],
+            "layout": "IPY_MODEL_3eaf53e3d6674527aaca6f4aeda71f7d"
+          }
+        },
+        "fef8c84b911a419c882ae7c0d8b6a937": {
           "model_module": "@jupyter-widgets/base",
-          "model_name": "LayoutModel",
           "model_module_version": "1.2.0",
+          "model_name": "LayoutModel",
           "state": {
             "_model_module": "@jupyter-widgets/base",
             "_model_module_version": "1.2.0",
@@ -3104,25 +3117,10 @@
             "visibility": null,
             "width": null
           }
-        },
-        "9c8a77cd6bce4d9a8e3c81714cad6964": {
-          "model_module": "@jupyter-widgets/controls",
-          "model_name": "DescriptionStyleModel",
-          "model_module_version": "1.5.0",
-          "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": ""
-          }
         }
       }
     }
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}