diff --git "a/notebook/GPT2_DrugsCom_DepressionReviews_FineTuning.ipynb" "b/notebook/GPT2_DrugsCom_DepressionReviews_FineTuning.ipynb"
new file mode 100644--- /dev/null
+++ "b/notebook/GPT2_DrugsCom_DepressionReviews_FineTuning.ipynb"
@@ -0,0 +1,8340 @@
+{
+  "cells": [
+    {
+      "cell_type": "markdown",
+      "source": [
+        "# Fine-Tuning GPT-2 on Drugs.com Reviews for Depression\n",
+        "\n",
+        "**Author**: Zakia Salod\n",
+        "\n",
+        "**Affiliation**: University of KwaZulu-Natal (UKZN), Durban, South Africa\n",
+        "\n",
+        "**Contact**: zakia.salod@gmail.com\n",
+        "\n",
+        "**Machine Used**: Google Colab T4 GPU\n",
+        "\n",
+        "**Last Updated**: 9 December 2023\n",
+        "\n",
+        "**Description**:\n",
+        "This notebook demonstrates the process of fine-tuning the GPT-2 model using the Drugs.com reviews dataset, focusing specifically on reviews related to depression. The notebook covers the steps of loading and preprocessing the dataset, initializing and preparing the GPT-2 model for fine-tuning, and the fine-tuning process itself. Post-training, the notebook illustrates how to save and push the fine-tuned model to the Hugging Face Hub. This exercise aims to enhance GPT-2's ability to generate more relevant and insightful text related to depression drug reviews.\n",
+        "\n",
+        "\n",
+        "**License**:\n",
+        "This work is licensed under the [Apache 2.0 License](https://www.apache.org/licenses/LICENSE-2.0). Free for educational and research use.\n",
+        "\n"
+      ],
+      "metadata": {
+        "id": "SlML_A8WXRKX"
+      }
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "kJK4WrJB4JR1"
+      },
+      "source": [
+        "## STEP 1: SETTING UP THE ENVIRONMENT"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "zzrRM4P14M7-"
+      },
+      "source": [
+        "### Load Necessary Libraries"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "zbzxLSs_tKQ4"
+      },
+      "outputs": [],
+      "source": [
+        "# Enable automatic module reloading to reflect changes in external .py files\n",
+        "%load_ext autoreload\n",
+        "# Reload all modules before executing code, keeping modules up-to-date\n",
+        "%autoreload 2"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "### Install Required Packages"
+      ],
+      "metadata": {
+        "id": "fq1xfyKlf_Ea"
+      }
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "Bv85wGPxto3O",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "dab3659e-59ab-41a4-9a70-838773c85c5a"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Collecting git+https://github.com/huggingface/transformers\n",
+            "  Cloning https://github.com/huggingface/transformers to /tmp/pip-req-build-44ijr861\n",
+            "  Running command git clone --filter=blob:none --quiet https://github.com/huggingface/transformers /tmp/pip-req-build-44ijr861\n",
+            "  Resolved https://github.com/huggingface/transformers to commit df5c5c62ae253055336f5bb0828ca8e3e15ab6bd\n",
+            "  Installing build dependencies ... \u001b[?25l\u001b[?25hdone\n",
+            "  Getting requirements to build wheel ... \u001b[?25l\u001b[?25hdone\n",
+            "  Preparing metadata (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n",
+            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (3.13.1)\n",
+            "Requirement already satisfied: huggingface-hub<1.0,>=0.19.3 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (0.19.4)\n",
+            "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (1.23.5)\n",
+            "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (23.2)\n",
+            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (6.0.1)\n",
+            "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (2023.6.3)\n",
+            "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (2.31.0)\n",
+            "Requirement already satisfied: tokenizers<0.19,>=0.14 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (0.15.0)\n",
+            "Requirement already satisfied: safetensors>=0.3.1 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (0.4.1)\n",
+            "Requirement already satisfied: tqdm>=4.27 in /usr/local/lib/python3.10/dist-packages (from transformers==4.36.0.dev0) (4.66.1)\n",
+            "Requirement already satisfied: fsspec>=2023.5.0 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.19.3->transformers==4.36.0.dev0) (2023.6.0)\n",
+            "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.19.3->transformers==4.36.0.dev0) (4.5.0)\n",
+            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->transformers==4.36.0.dev0) (3.3.2)\n",
+            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->transformers==4.36.0.dev0) (3.6)\n",
+            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->transformers==4.36.0.dev0) (2.0.7)\n",
+            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->transformers==4.36.0.dev0) (2023.11.17)\n",
+            "Building wheels for collected packages: transformers\n",
+            "  Building wheel for transformers (pyproject.toml) ... \u001b[?25l\u001b[?25hdone\n",
+            "  Created wheel for transformers: filename=transformers-4.36.0.dev0-py3-none-any.whl size=8195433 sha256=79b5b0d64790d529619e06ffc24ad2f2a2692b0a7d84d20a1e0c4dae56550d23\n",
+            "  Stored in directory: /tmp/pip-ephem-wheel-cache-ogqdggii/wheels/c0/14/d6/6c9a5582d2ac191ec0a483be151a4495fe1eb2a6706ca49f1b\n",
+            "Successfully built transformers\n",
+            "Installing collected packages: transformers\n",
+            "  Attempting uninstall: transformers\n",
+            "    Found existing installation: transformers 4.35.2\n",
+            "    Uninstalling transformers-4.35.2:\n",
+            "      Successfully uninstalled transformers-4.35.2\n",
+            "Successfully installed transformers-4.36.0.dev0\n"
+          ]
+        }
+      ],
+      "source": [
+        "!pip install git+https://github.com/huggingface/transformers"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "ZTZrScZY6NB1",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "5342e849-426d-437f-a9c2-b75adfbcb4bf"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Collecting datasets\n",
+            "  Downloading datasets-2.15.0-py3-none-any.whl (521 kB)\n",
+            "\u001b[?25l     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m0.0/521.2 kB\u001b[0m \u001b[31m?\u001b[0m eta \u001b[36m-:--:--\u001b[0m\r\u001b[2K     \u001b[91m━━━━━━━━━━━\u001b[0m\u001b[90m╺\u001b[0m\u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m143.4/521.2 kB\u001b[0m \u001b[31m4.6 MB/s\u001b[0m eta \u001b[36m0:00:01\u001b[0m\r\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m521.2/521.2 kB\u001b[0m \u001b[31m8.9 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hCollecting evaluate\n",
+            "  Downloading evaluate-0.4.1-py3-none-any.whl (84 kB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m84.1/84.1 kB\u001b[0m \u001b[31m11.3 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hRequirement already satisfied: torch in /usr/local/lib/python3.10/dist-packages (2.1.0+cu118)\n",
+            "Collecting wandb\n",
+            "  Downloading wandb-0.16.1-py3-none-any.whl (2.1 MB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m2.1/2.1 MB\u001b[0m \u001b[31m31.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hRequirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from datasets) (1.23.5)\n",
+            "Requirement already satisfied: pyarrow>=8.0.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (9.0.0)\n",
+            "Collecting pyarrow-hotfix (from datasets)\n",
+            "  Downloading pyarrow_hotfix-0.6-py3-none-any.whl (7.9 kB)\n",
+            "Collecting dill<0.3.8,>=0.3.0 (from datasets)\n",
+            "  Downloading dill-0.3.7-py3-none-any.whl (115 kB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m115.3/115.3 kB\u001b[0m \u001b[31m15.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hRequirement already satisfied: pandas in /usr/local/lib/python3.10/dist-packages (from datasets) (1.5.3)\n",
+            "Requirement already satisfied: requests>=2.19.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (2.31.0)\n",
+            "Requirement already satisfied: tqdm>=4.62.1 in /usr/local/lib/python3.10/dist-packages (from datasets) (4.66.1)\n",
+            "Requirement already satisfied: xxhash in /usr/local/lib/python3.10/dist-packages (from datasets) (3.4.1)\n",
+            "Collecting multiprocess (from datasets)\n",
+            "  Downloading multiprocess-0.70.15-py310-none-any.whl (134 kB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m134.8/134.8 kB\u001b[0m \u001b[31m18.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hRequirement already satisfied: fsspec[http]<=2023.10.0,>=2023.1.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (2023.6.0)\n",
+            "Requirement already satisfied: aiohttp in /usr/local/lib/python3.10/dist-packages (from datasets) (3.9.1)\n",
+            "Requirement already satisfied: huggingface-hub>=0.18.0 in /usr/local/lib/python3.10/dist-packages (from datasets) (0.19.4)\n",
+            "Requirement already satisfied: packaging in /usr/local/lib/python3.10/dist-packages (from datasets) (23.2)\n",
+            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from datasets) (6.0.1)\n",
+            "Collecting responses<0.19 (from evaluate)\n",
+            "  Downloading responses-0.18.0-py3-none-any.whl (38 kB)\n",
+            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch) (3.13.1)\n",
+            "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch) (4.5.0)\n",
+            "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch) (1.12)\n",
+            "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch) (3.2.1)\n",
+            "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch) (3.1.2)\n",
+            "Requirement already satisfied: triton==2.1.0 in /usr/local/lib/python3.10/dist-packages (from torch) (2.1.0)\n",
+            "Requirement already satisfied: Click!=8.0.0,>=7.1 in /usr/local/lib/python3.10/dist-packages (from wandb) (8.1.7)\n",
+            "Collecting GitPython!=3.1.29,>=1.0.0 (from wandb)\n",
+            "  Downloading GitPython-3.1.40-py3-none-any.whl (190 kB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m190.6/190.6 kB\u001b[0m \u001b[31m23.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hRequirement already satisfied: psutil>=5.0.0 in /usr/local/lib/python3.10/dist-packages (from wandb) (5.9.5)\n",
+            "Collecting sentry-sdk>=1.0.0 (from wandb)\n",
+            "  Downloading sentry_sdk-1.38.0-py2.py3-none-any.whl (252 kB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m252.8/252.8 kB\u001b[0m \u001b[31m28.6 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hCollecting docker-pycreds>=0.4.0 (from wandb)\n",
+            "  Downloading docker_pycreds-0.4.0-py2.py3-none-any.whl (9.0 kB)\n",
+            "Collecting setproctitle (from wandb)\n",
+            "  Downloading setproctitle-1.3.3-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (30 kB)\n",
+            "Requirement already satisfied: setuptools in /usr/local/lib/python3.10/dist-packages (from wandb) (67.7.2)\n",
+            "Requirement already satisfied: appdirs>=1.4.3 in /usr/local/lib/python3.10/dist-packages (from wandb) (1.4.4)\n",
+            "Requirement already satisfied: protobuf!=4.21.0,<5,>=3.19.0 in /usr/local/lib/python3.10/dist-packages (from wandb) (3.20.3)\n",
+            "Requirement already satisfied: six>=1.4.0 in /usr/local/lib/python3.10/dist-packages (from docker-pycreds>=0.4.0->wandb) (1.16.0)\n",
+            "Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (23.1.0)\n",
+            "Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (6.0.4)\n",
+            "Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.9.3)\n",
+            "Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.4.0)\n",
+            "Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (1.3.1)\n",
+            "Requirement already satisfied: async-timeout<5.0,>=4.0 in /usr/local/lib/python3.10/dist-packages (from aiohttp->datasets) (4.0.3)\n",
+            "Collecting gitdb<5,>=4.0.1 (from GitPython!=3.1.29,>=1.0.0->wandb)\n",
+            "  Downloading gitdb-4.0.11-py3-none-any.whl (62 kB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m62.7/62.7 kB\u001b[0m \u001b[31m9.5 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hRequirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->datasets) (3.3.2)\n",
+            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->datasets) (3.6)\n",
+            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->datasets) (2.0.7)\n",
+            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests>=2.19.0->datasets) (2023.11.17)\n",
+            "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch) (2.1.3)\n",
+            "Requirement already satisfied: python-dateutil>=2.8.1 in /usr/local/lib/python3.10/dist-packages (from pandas->datasets) (2.8.2)\n",
+            "Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.10/dist-packages (from pandas->datasets) (2023.3.post1)\n",
+            "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch) (1.3.0)\n",
+            "Collecting smmap<6,>=3.0.1 (from gitdb<5,>=4.0.1->GitPython!=3.1.29,>=1.0.0->wandb)\n",
+            "  Downloading smmap-5.0.1-py3-none-any.whl (24 kB)\n",
+            "Installing collected packages: smmap, setproctitle, sentry-sdk, pyarrow-hotfix, docker-pycreds, dill, responses, multiprocess, gitdb, GitPython, wandb, datasets, evaluate\n",
+            "Successfully installed GitPython-3.1.40 datasets-2.15.0 dill-0.3.7 docker-pycreds-0.4.0 evaluate-0.4.1 gitdb-4.0.11 multiprocess-0.70.15 pyarrow-hotfix-0.6 responses-0.18.0 sentry-sdk-1.38.0 setproctitle-1.3.3 smmap-5.0.1 wandb-0.16.1\n"
+          ]
+        }
+      ],
+      "source": [
+        "!pip install datasets evaluate torch wandb"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "CHbwWz_gH1qc",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "18979ebb-9fce-4d87-dccf-ddae7d9df25c"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Collecting accelerate\n",
+            "  Downloading accelerate-0.25.0-py3-none-any.whl (265 kB)\n",
+            "\u001b[2K     \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m265.7/265.7 kB\u001b[0m \u001b[31m5.2 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n",
+            "\u001b[?25hRequirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from accelerate) (1.23.5)\n",
+            "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from accelerate) (23.2)\n",
+            "Requirement already satisfied: psutil in /usr/local/lib/python3.10/dist-packages (from accelerate) (5.9.5)\n",
+            "Requirement already satisfied: pyyaml in /usr/local/lib/python3.10/dist-packages (from accelerate) (6.0.1)\n",
+            "Requirement already satisfied: torch>=1.10.0 in /usr/local/lib/python3.10/dist-packages (from accelerate) (2.1.0+cu118)\n",
+            "Requirement already satisfied: huggingface-hub in /usr/local/lib/python3.10/dist-packages (from accelerate) (0.19.4)\n",
+            "Requirement already satisfied: safetensors>=0.3.1 in /usr/local/lib/python3.10/dist-packages (from accelerate) (0.4.1)\n",
+            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate) (3.13.1)\n",
+            "Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate) (4.5.0)\n",
+            "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate) (1.12)\n",
+            "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate) (3.2.1)\n",
+            "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate) (3.1.2)\n",
+            "Requirement already satisfied: fsspec in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate) (2023.6.0)\n",
+            "Requirement already satisfied: triton==2.1.0 in /usr/local/lib/python3.10/dist-packages (from torch>=1.10.0->accelerate) (2.1.0)\n",
+            "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from huggingface-hub->accelerate) (2.31.0)\n",
+            "Requirement already satisfied: tqdm>=4.42.1 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub->accelerate) (4.66.1)\n",
+            "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch>=1.10.0->accelerate) (2.1.3)\n",
+            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate) (3.3.2)\n",
+            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate) (3.6)\n",
+            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate) (2.0.7)\n",
+            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->huggingface-hub->accelerate) (2023.11.17)\n",
+            "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch>=1.10.0->accelerate) (1.3.0)\n",
+            "Installing collected packages: accelerate\n",
+            "Successfully installed accelerate-0.25.0\n"
+          ]
+        }
+      ],
+      "source": [
+        "!pip install accelerate -U"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "8ClR-_ktH6-3",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "1bc58861-c84b-4b0d-bd02-87f8fb39fbf0"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Requirement already satisfied: transformers[torch] in /usr/local/lib/python3.10/dist-packages (4.36.0.dev0)\n",
+            "Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (3.13.1)\n",
+            "Requirement already satisfied: huggingface-hub<1.0,>=0.19.3 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (0.19.4)\n",
+            "Requirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (1.23.5)\n",
+            "Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (23.2)\n",
+            "Requirement already satisfied: pyyaml>=5.1 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (6.0.1)\n",
+            "Requirement already satisfied: regex!=2019.12.17 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (2023.6.3)\n",
+            "Requirement already satisfied: requests in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (2.31.0)\n",
+            "Requirement already satisfied: tokenizers<0.19,>=0.14 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (0.15.0)\n",
+            "Requirement already satisfied: safetensors>=0.3.1 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (0.4.1)\n",
+            "Requirement already satisfied: tqdm>=4.27 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (4.66.1)\n",
+            "Requirement already satisfied: torch!=1.12.0,>=1.10 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (2.1.0+cu118)\n",
+            "Requirement already satisfied: accelerate>=0.21.0 in /usr/local/lib/python3.10/dist-packages (from transformers[torch]) (0.25.0)\n",
+            "Requirement already satisfied: psutil in /usr/local/lib/python3.10/dist-packages (from accelerate>=0.21.0->transformers[torch]) (5.9.5)\n",
+            "Requirement already satisfied: fsspec>=2023.5.0 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.19.3->transformers[torch]) (2023.6.0)\n",
+            "Requirement already satisfied: typing-extensions>=3.7.4.3 in /usr/local/lib/python3.10/dist-packages (from huggingface-hub<1.0,>=0.19.3->transformers[torch]) (4.5.0)\n",
+            "Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch!=1.12.0,>=1.10->transformers[torch]) (1.12)\n",
+            "Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch!=1.12.0,>=1.10->transformers[torch]) (3.2.1)\n",
+            "Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch!=1.12.0,>=1.10->transformers[torch]) (3.1.2)\n",
+            "Requirement already satisfied: triton==2.1.0 in /usr/local/lib/python3.10/dist-packages (from torch!=1.12.0,>=1.10->transformers[torch]) (2.1.0)\n",
+            "Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests->transformers[torch]) (3.3.2)\n",
+            "Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests->transformers[torch]) (3.6)\n",
+            "Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests->transformers[torch]) (2.0.7)\n",
+            "Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests->transformers[torch]) (2023.11.17)\n",
+            "Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch!=1.12.0,>=1.10->transformers[torch]) (2.1.3)\n",
+            "Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch!=1.12.0,>=1.10->transformers[torch]) (1.3.0)\n"
+          ]
+        }
+      ],
+      "source": [
+        "!pip install transformers[torch]"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "source": [
+        "### Import Necessary Libraries"
+      ],
+      "metadata": {
+        "id": "GXDAGYt9gFFM"
+      }
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "ZPQo3nOFtpX-"
+      },
+      "outputs": [],
+      "source": [
+        "import numpy as np\n",
+        "from datasets import load_dataset\n",
+        "from datasets import concatenate_datasets\n",
+        "from transformers import GPT2Tokenizer\n",
+        "from transformers import GPT2LMHeadModel\n",
+        "import html\n",
+        "import re\n",
+        "from collections import Counter\n",
+        "import pandas as pd"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "CAc15TpfT0m6"
+      },
+      "source": [
+        "### Initialize Weights & Biases for Tracking"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "gu1-uIyZT179"
+      },
+      "outputs": [],
+      "source": [
+        "# Import Weights & Biases for experiment tracking\n",
+        "import wandb"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "FAcoX4eniLWc",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 211
+        },
+        "outputId": "5726f50d-2232-44bc-c4fb-a8cd9c8fe69f"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "<IPython.core.display.Javascript object>"
+            ],
+            "application/javascript": [
+              "\n",
+              "        window._wandbApiKey = new Promise((resolve, reject) => {\n",
+              "            function loadScript(url) {\n",
+              "            return new Promise(function(resolve, reject) {\n",
+              "                let newScript = document.createElement(\"script\");\n",
+              "                newScript.onerror = reject;\n",
+              "                newScript.onload = resolve;\n",
+              "                document.body.appendChild(newScript);\n",
+              "                newScript.src = url;\n",
+              "            });\n",
+              "            }\n",
+              "            loadScript(\"https://cdn.jsdelivr.net/npm/postmate/build/postmate.min.js\").then(() => {\n",
+              "            const iframe = document.createElement('iframe')\n",
+              "            iframe.style.cssText = \"width:0;height:0;border:none\"\n",
+              "            document.body.appendChild(iframe)\n",
+              "            const handshake = new Postmate({\n",
+              "                container: iframe,\n",
+              "                url: 'https://wandb.ai/authorize'\n",
+              "            });\n",
+              "            const timeout = setTimeout(() => reject(\"Couldn't auto authenticate\"), 5000)\n",
+              "            handshake.then(function(child) {\n",
+              "                child.on('authorize', data => {\n",
+              "                    clearTimeout(timeout)\n",
+              "                    resolve(data)\n",
+              "                });\n",
+              "            });\n",
+              "            })\n",
+              "        });\n",
+              "    "
+            ]
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "stream",
+          "name": "stderr",
+          "text": [
+            "\u001b[34m\u001b[1mwandb\u001b[0m: Logging into wandb.ai. (Learn how to deploy a W&B server locally: https://wandb.me/wandb-server)\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: You can find your API key in your browser here: https://wandb.ai/authorize\n",
+            "wandb: Paste an API key from your profile and hit enter, or press ctrl+c to quit:"
+          ]
+        },
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            " ··········\n"
+          ]
+        },
+        {
+          "output_type": "stream",
+          "name": "stderr",
+          "text": [
+            "\u001b[34m\u001b[1mwandb\u001b[0m: Appending key for api.wandb.ai to your netrc file: /root/.netrc\n"
+          ]
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "<IPython.core.display.HTML object>"
+            ],
+            "text/html": [
+              "Tracking run with wandb version 0.16.1"
+            ]
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "<IPython.core.display.HTML object>"
+            ],
+            "text/html": [
+              "Run data is saved locally in <code>/content/wandb/run-20231209_123209-kfb5cx1j</code>"
+            ]
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "<IPython.core.display.HTML object>"
+            ],
+            "text/html": [
+              "Syncing run <strong><a href='https://wandb.ai/team-zakia/gpt2-drugscom_depression_reviews/runs/kfb5cx1j' target=\"_blank\">jumping-energy-8</a></strong> to <a href='https://wandb.ai/team-zakia/gpt2-drugscom_depression_reviews' target=\"_blank\">Weights & Biases</a> (<a href='https://wandb.me/run' target=\"_blank\">docs</a>)<br/>"
+            ]
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "<IPython.core.display.HTML object>"
+            ],
+            "text/html": [
+              " View project at <a href='https://wandb.ai/team-zakia/gpt2-drugscom_depression_reviews' target=\"_blank\">https://wandb.ai/team-zakia/gpt2-drugscom_depression_reviews</a>"
+            ]
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "<IPython.core.display.HTML object>"
+            ],
+            "text/html": [
+              " View run at <a href='https://wandb.ai/team-zakia/gpt2-drugscom_depression_reviews/runs/kfb5cx1j' target=\"_blank\">https://wandb.ai/team-zakia/gpt2-drugscom_depression_reviews/runs/kfb5cx1j</a>"
+            ]
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "execute_result",
+          "data": {
+            "text/html": [
+              "<button onClick=\"this.nextSibling.style.display='block';this.style.display='none';\">Display W&B run</button><iframe src='https://wandb.ai/team-zakia/gpt2-drugscom_depression_reviews/runs/kfb5cx1j?jupyter=true' style='border:none;width:100%;height:420px;display:none;'></iframe>"
+            ],
+            "text/plain": [
+              "<wandb.sdk.wandb_run.Run at 0x791d9a29f670>"
+            ]
+          },
+          "metadata": {},
+          "execution_count": 8
+        }
+      ],
+      "source": [
+        "# Initialize wandb for tracking and visualizing the training\n",
+        "wandb.init(project=\"gpt2-drugscom_depression_reviews\")"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "LGYKu-eF4ZvJ"
+      },
+      "source": [
+        "## STEP 2: LOAD AND PRE-PROCESS DRUG REVIEWS DATASET"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "v_FNuZmvtsY1"
+      },
+      "outputs": [],
+      "source": [
+        "dataset_name = \"Zakia/drugscom_reviews\""
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "DyK71VIut7kf",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 241,
+          "referenced_widgets": [
+            "b45c8c73698048aeaee4d34b28d6a27a",
+            "90e79e1e73ae489199ff3811dca9adae",
+            "e59b83647a7f483da05b37429322c469",
+            "66ec2664647846d8b4da85fee9e32df1",
+            "6b6eec8ea25445a6bc24b88778e1edbf",
+            "f2722d9ef4484a2a90ca9d6319d04345",
+            "9cc770f5d35e421cb07b44c9f7e2a730",
+            "2c52eb43c73e4c7f8831bd6f71809570",
+            "02587cf05a5547c4ba874c34bf4cec4b",
+            "0707b92bf1a84a9384beba55789ce6ba",
+            "57e880d55b0548a0b3a0f61ccdcc930f",
+            "10311a17cdf645f69a3edd08ca2cad62",
+            "a6000f68e93d4bf1a343629032dcef04",
+            "ac869d78cfe146879a2494579c77277d",
+            "2e35f836eaed419a8431e0c68e8d2cc1",
+            "a3d694bb06194b138a7abdbf199a1d81",
+            "5bdaf9e1d6da4b808b2a704b3c8d973b",
+            "2e5a51a0fb4c4249b55d8af4b0506eca",
+            "b4fcbf969be2427b92046a1f528db2a6",
+            "43c31088bf444fd39578f2fe630dc46e",
+            "3f7cb58407064ebaa1ec3943309c95c3",
+            "ce17ffc4033b4866b48d5e61467db6c3",
+            "0ffc9ba03f924bc0ad48fa82c2e711e7",
+            "0abb262f82df4d728a8f05bf992a5ec3",
+            "09e3afd0ba9e4b5298d741ce825fdbfb",
+            "f069002ce14648c3b0269ae3e377879c",
+            "1b6b3d597a024cfb9a15573cea12891e",
+            "e48c988e59b8462babda6b5173a337a6",
+            "43f34a6373a7449a83d86705351d4957",
+            "3794962a26714723a2b7d345336db7ec",
+            "19ee2670c44b46e2906b21a73b27ba88",
+            "16a81359fc4a40dbb83d185c1db81ebf",
+            "8970468c315b413abb6541205fc3174d",
+            "84ba5a9fb669445d88a8f18c86986a65",
+            "74e6479c15a843c196b26eae181f3b50",
+            "24f3bb13e552446ca3dbf1d5acdd306e",
+            "adf7780fbbed4baf90f08b113553e0a3",
+            "59b48ccd44734f2fb869291b1c52a56b",
+            "bff2b32d8b624af98814c950bf2934d9",
+            "8b5cf7a76357475d8156886940ad2790",
+            "a8a19e8b09a84ea9a29a6dd8312d2110",
+            "654b39a5572f42bfa93ba73c4b251fbd",
+            "76464bf094c74cd68f3fb5e9975ff56e",
+            "1a3a890317814810834c117ffe62f390",
+            "a6687ff1f33a42988536dc04ccf824ad",
+            "6e79a05a0f6f4e4c8b0ce54d47180df1",
+            "c7f71309b9a14d9b85a9953f1c5ef919",
+            "ad491e5e0a8d4106b6942f20206081f4",
+            "a08bd346d672441994b14340f57fa298",
+            "d9166639b16c423fa902ef9a23a6582f",
+            "0e564d1c6ae8447fb431032ee5d64199",
+            "c22ad507940c46d0bf34dc5efe8ffee3",
+            "38331b41172b47fa96b2b7950dd4f94b",
+            "15038217fa3c4c43990aeb222792feed",
+            "8bad5a126935481497fdb16ad2dacbb5",
+            "28b5d0115eb44b9483c9f7d345d644e1",
+            "5aa2e3b0f8024725a260898904739acb",
+            "bdb27820a5204529987b9e2df414df78",
+            "f4f30d7e47864e4c9515ef6195661782",
+            "c4230c8b74d540afa5d2e885310dbf6b",
+            "bf9000b17de04dd1bd522f94a3eb1ed9",
+            "97411c1fec18481cbd5bcab1e794cc7f",
+            "875e1cbc8e634cbaa00a22ff3c9d62f8",
+            "52d6693766714e91b126d9180dd03346",
+            "f247e069bb4147e0b6019d66d851ede9",
+            "6a509c81bfff4f30bc7d2de029bdc93c",
+            "36a765c57fc349eab616355610980324",
+            "210adeaaf64e40db8323e085794e7d7a",
+            "b8b1a3dfbb984ba08b1c862b26af8dbe",
+            "304df8debe3f4662a4171221d669798e",
+            "266e9676bce140b69ad73456851ff499",
+            "9b0b095167de49f4bc5efb5502b3d41c",
+            "af16b06528824cc9b0ce63fc4b4738ba",
+            "daa61ca2c88c49be9a115fb81f01caac",
+            "5edea6130ae64675a5aecbbaffcc5821",
+            "1b55987264384ee3bd90ba9e97842b61",
+            "2a4ea7765a984268a58766f089744cdb"
+          ]
+        },
+        "outputId": "bd5d0427-c53e-4168-efba-44f64eadee42"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading readme:   0%|          | 0.00/6.72k [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "b45c8c73698048aeaee4d34b28d6a27a"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading data files:   0%|          | 0/2 [00:00<?, ?it/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "10311a17cdf645f69a3edd08ca2cad62"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/84.3M [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "0ffc9ba03f924bc0ad48fa82c2e711e7"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Downloading data:   0%|          | 0.00/28.1M [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "84ba5a9fb669445d88a8f18c86986a65"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Extracting data files:   0%|          | 0/2 [00:00<?, ?it/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "a6687ff1f33a42988536dc04ccf824ad"
+            }
+          },
+          "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": "28b5d0115eb44b9483c9f7d345d644e1"
+            }
+          },
+          "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": "36a765c57fc349eab616355610980324"
+            }
+          },
+          "metadata": {}
+        }
+      ],
+      "source": [
+        "ds = load_dataset(dataset_name, split=\"train\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "mIIzKwC3t9k_",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 49,
+          "referenced_widgets": [
+            "35b794e3143a43b9867050806d00d7c2",
+            "c0aabbfff4ca48609969258d77daf36c",
+            "a656a7cec90141f480a2d447e271edd4",
+            "fbd866b694004b478d12d7c8d621c563",
+            "a67a7144c6674c8eb119e5898b2d0490",
+            "4fe2cf6e63824e74b008998b6378bf01",
+            "29c67f208b3f4b90a38bba90aa369dcd",
+            "db9cdd8b18f24795ba9bd91a875ce448",
+            "5abc8caabb924c958c213348fde64848",
+            "afe1ff9c1d014590bba254ee2285c5c4",
+            "0c4015e83ef142bfb41abfde20a3fa05"
+          ]
+        },
+        "outputId": "f3b93127-7a4b-42d9-b62a-3e5be2fafda1"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Filter:   0%|          | 0/161297 [00:00<?, ? examples/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "35b794e3143a43b9867050806d00d7c2"
+            }
+          },
+          "metadata": {}
+        }
+      ],
+      "source": [
+        "# Filter out rows with missing drugName, condition, or review\n",
+        "ds = ds.filter(lambda x: all([x.get(\"drugName\"), x.get(\"condition\"), x.get(\"review\")]))"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "RgLtwYudt_-y",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 49,
+          "referenced_widgets": [
+            "361e2c414ad942f5b8c92a5dcd3b1774",
+            "dd3dbc5532264fb7881f11b0f3554b38",
+            "48f3c24e3f384907a0e359b92e952914",
+            "caeba295e0a84b21b6bece98ab886e1f",
+            "0651146886cc4d969d9f9a2ed6d18c8a",
+            "18453188685544d69ffc475599cb57db",
+            "5cdaa0d7536b42cbaa1025dcdf4c911f",
+            "1193555c580545759bfcbf6a1913a5d5",
+            "77ccf09db8ad4452ac0b307081b68b05",
+            "f859e097073941d5b6baf569a731f396",
+            "cb19c04f36f84e5cb1d96dc520c6566d"
+          ]
+        },
+        "outputId": "ce4512ed-d4db-4579-e166-f5d5921ffd73"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Filter:   0%|          | 0/160398 [00:00<?, ? examples/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "361e2c414ad942f5b8c92a5dcd3b1774"
+            }
+          },
+          "metadata": {}
+        }
+      ],
+      "source": [
+        "# Filter the dataset for the condition 'Depression'\n",
+        "ds = ds.filter(lambda example: example[\"condition\"] == \"Depression\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# Get the number of records\n",
+        "num_records = ds.num_rows\n",
+        "print(f\"Number of records with Depression condition: {num_records}\")"
+      ],
+      "metadata": {
+        "id": "JfVTTUk63kZQ",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "1e1e08dd-b055-44a5-940b-f6dad88f1f1e"
+      },
+      "execution_count": null,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Number of records with Depression condition: 9069\n"
+          ]
+        }
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "2GmihTdSuMPF"
+      },
+      "outputs": [],
+      "source": [
+        "# Function to clean review text\n",
+        "def clean_review(text):\n",
+        "    # Check if the text is a string\n",
+        "    if not isinstance(text, str):\n",
+        "      return \"\"  # Return an empty string if the input is not a string\n",
+        "    text = html.unescape(text)  # Decode HTML entities\n",
+        "    text = re.sub(r'\"', '', text)  # Remove quotes\n",
+        "    text = re.sub(r'<.*?>', '', text)  # Remove HTML tags\n",
+        "    return text"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "jSEt0xnfuSQu"
+      },
+      "outputs": [],
+      "source": [
+        "# Clean the reviews of the train set\n",
+        "# Apply the clean_review function in a batched manner\n",
+        "def clean_reviews(batch):\n",
+        "    # Apply clean_review to each review in the batch and return the modified batch\n",
+        "    return {\"review\": [clean_review(review) for review in batch[\"review\"]]}"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "llYAOpmpuTad",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 49,
+          "referenced_widgets": [
+            "48f3b3e586f74174a0232079ec543e2c",
+            "57403d31cdb64f11add87566d4f5a325",
+            "f9e6f05aeb904395a5d5a8a3ec3b9756",
+            "5b787cdd23f44cb8805e055ccffdfe40",
+            "e00ea9a3b8cc433a8dffc1cf9d99f609",
+            "c6cae5d28bd9480eb1eb5642132d4dd5",
+            "90e73a35476f4e74aba7c02f329a3536",
+            "dea3da72e6d946029a3760a594356701",
+            "5c7f4c3d4ebc47a6ba27aeed52a89add",
+            "23396629e70847a28c247157d27587e7",
+            "2262d8d8318e453a92e4ae9b37fa484c"
+          ]
+        },
+        "outputId": "3815254b-c672-41ad-8227-5a925d00d4d7"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Map:   0%|          | 0/9069 [00:00<?, ? examples/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "48f3b3e586f74174a0232079ec543e2c"
+            }
+          },
+          "metadata": {}
+        }
+      ],
+      "source": [
+        "ds = ds.map(clean_reviews, batched=True)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# Get the number of records\n",
+        "num_records = ds.num_rows\n",
+        "print(f\"Number of records with Depression condition: {num_records}\")"
+      ],
+      "metadata": {
+        "id": "manmVhe43meU",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "dbf58348-48dc-4a04-a784-510533f88fe5"
+      },
+      "execution_count": null,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Number of records with Depression condition: 9069\n"
+          ]
+        }
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "QYzV-JbK0hg_"
+      },
+      "source": [
+        "## STEP 3: PREPARE THE GPT-2 MODEL FOR FINE-TUNING"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "tyRs25rj0uZq",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 145,
+          "referenced_widgets": [
+            "97478b946a2243ae8fb5e2dbc939fecd",
+            "b468cc474d964795aa92e609d8f49c14",
+            "c208ce8f7548438a886309c3a1d89a46",
+            "3f11e4cb4ea2467685187f86c6ad1259",
+            "78db2d61bf444938bd55d74ee2962bc7",
+            "1c31184a73fa4ded96c3fc98aa98e4dd",
+            "feb1dec1d9684a8ebc23a97372cd836e",
+            "6a1f524d208041bc9f632658154e9604",
+            "aa0834b1a9774b039c81c30abd225c75",
+            "e6bf66b3ae9246af8dfb901f44114036",
+            "a3606f6637814cadaddeec5b809a2031",
+            "c14073600faa4276baabf68a8463d203",
+            "86eeb0998e1f4d79bfe1d82cb7dec973",
+            "5d65c6687d87417193211910daf341c9",
+            "6cf38e7be70f4297a4a4a76afb00f5eb",
+            "f3530c618bed4888a151573bcec3df84",
+            "d28f426d69a94954923823dbf6bca790",
+            "b4cc09eaece64d81a60cd88a4dfd8327",
+            "1194b9e87bf24c8190a7114527f153cb",
+            "2b42c838b36445c28ad2c1d5802b94a8",
+            "37453894d05741b5b76fbb25e00f3f4f",
+            "4e19eea754254320b0c64b7518494ca8",
+            "f3be5e77bcc14067a0b8333488b6989a",
+            "063a528402a041c58ca72450e8f94a47",
+            "f2b64c328fee41b5b137fd97fca75a63",
+            "fa00fe6d3f074db9965be53e4694a830",
+            "fbe2c79add1e4125b9e8bc1b24e2bc7f",
+            "a6e378877e2146948f3e412bd31252ed",
+            "681923cbf6874204898918586d4d15e5",
+            "5f9839dab22a475f95c0547fc5035a67",
+            "14a56ddc2f654e31b599bfbfc967d879",
+            "7d3128fda8c844b1bc84fadd2997a09a",
+            "da3bc0b8f8644bd2834089beb5aff4d7",
+            "80eba5b6545948269049d0b804fc8f40",
+            "ac32cb2229e541bf8aef1f7dd2ada74b",
+            "a796211c642c4840a2822833d00f3ac0",
+            "cc3831986c9a41099813df38119f8605",
+            "faf882727a9e442ca422ee63d02357c9",
+            "88ecab3b211e4cf69a76bb7b83ae0f7f",
+            "2294c3691218465aba561ce60038a691",
+            "057e01cd640f4b6bb05dbce84b40002d",
+            "0733564e11614f3a9e4480222a58286d",
+            "6f8671899ef74ea48a78df36ace76266",
+            "21f9727629834203afd92e6896f22071"
+          ]
+        },
+        "outputId": "85d022b0-5b1e-45d3-f277-ebfeee182d8b"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "vocab.json:   0%|          | 0.00/1.04M [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "97478b946a2243ae8fb5e2dbc939fecd"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "merges.txt:   0%|          | 0.00/456k [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "c14073600faa4276baabf68a8463d203"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "tokenizer.json:   0%|          | 0.00/1.36M [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "f3be5e77bcc14067a0b8333488b6989a"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "config.json:   0%|          | 0.00/665 [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "80eba5b6545948269049d0b804fc8f40"
+            }
+          },
+          "metadata": {}
+        }
+      ],
+      "source": [
+        "# Initialize the tokenizer\n",
+        "tokenizer = GPT2Tokenizer.from_pretrained(\"gpt2\")"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "9kBUDFIZ8SCZ"
+      },
+      "outputs": [],
+      "source": [
+        "# Make sure the tokenizer recognizes the pad_token\n",
+        "tokenizer.pad_token = tokenizer.eos_token"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "PF-wYY0F03fw"
+      },
+      "outputs": [],
+      "source": [
+        "# Function to encode the texts\n",
+        "def encode(examples):\n",
+        "    return tokenizer(examples[\"review\"], truncation=True, padding=\"max_length\", max_length=512)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "18v0sBsc058q",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 49,
+          "referenced_widgets": [
+            "558e52210904446684aa3f3a52d03ada",
+            "9ddcd9da2ef34296829adce85e4f9c97",
+            "ddcf5f257a214b089f84ab0ed1320d53",
+            "22c72cc5f7104b2e82cb0363eed14243",
+            "153c517831cc4a9099bdfc1e3723ae02",
+            "ef41fb1aee914dfe83a10191ff488ac4",
+            "5e8d743b4c054f6b9fff9440522e9635",
+            "8199c19bba6944cd95276a43b7edfa8a",
+            "c46c0c2430ae4ea2a982425921282e13",
+            "537cb348212a4258af62b2cafdae8266",
+            "5e001995d8984ccd87b73d2d8a6c89ef"
+          ]
+        },
+        "outputId": "381fb94a-03ec-459c-d1da-bbc7e6a8fb45"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "Map:   0%|          | 0/9069 [00:00<?, ? examples/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "558e52210904446684aa3f3a52d03ada"
+            }
+          },
+          "metadata": {}
+        }
+      ],
+      "source": [
+        "# Apply the function to the dataset\n",
+        "ds = ds.map(encode, batched=True)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "C3oh7JQj1Hgq"
+      },
+      "outputs": [],
+      "source": [
+        "# Save the dataset to a file\n",
+        "with open(\"train_dataset.txt\", \"w\", encoding=\"utf-8\") as f:\n",
+        "    for example in ds:\n",
+        "        f.write(tokenizer.decode(example[\"input_ids\"]) + tokenizer.eos_token + \"\\n\")"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "GWgq1Sn64wYO"
+      },
+      "source": [
+        "## STEP 4: FINE-TUNING THE MODEL"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "LU84BWI31T_6",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "ffd82918-2314-41cd-c4ca-bb336a99c7d7"
+      },
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "2023-12-09 12:33:29.982409: E tensorflow/compiler/xla/stream_executor/cuda/cuda_dnn.cc:9342] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered\n",
+            "2023-12-09 12:33:29.982462: E tensorflow/compiler/xla/stream_executor/cuda/cuda_fft.cc:609] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered\n",
+            "2023-12-09 12:33:29.982496: E tensorflow/compiler/xla/stream_executor/cuda/cuda_blas.cc:1518] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered\n",
+            "2023-12-09 12:33:31.101611: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT\n",
+            "12/09/2023 12:33:39 - WARNING - __main__ - Process rank: 0, device: cuda:0, n_gpu: 1, distributed training: False, 16-bits training: False\n",
+            "12/09/2023 12:33:39 - INFO - __main__ - Training/evaluation parameters TrainingArguments(\n",
+            "_n_gpu=1,\n",
+            "adafactor=False,\n",
+            "adam_beta1=0.9,\n",
+            "adam_beta2=0.999,\n",
+            "adam_epsilon=1e-08,\n",
+            "auto_find_batch_size=False,\n",
+            "bf16=False,\n",
+            "bf16_full_eval=False,\n",
+            "data_seed=None,\n",
+            "dataloader_drop_last=False,\n",
+            "dataloader_num_workers=0,\n",
+            "dataloader_persistent_workers=False,\n",
+            "dataloader_pin_memory=True,\n",
+            "ddp_backend=None,\n",
+            "ddp_broadcast_buffers=None,\n",
+            "ddp_bucket_cap_mb=None,\n",
+            "ddp_find_unused_parameters=None,\n",
+            "ddp_timeout=1800,\n",
+            "debug=[],\n",
+            "deepspeed=None,\n",
+            "disable_tqdm=False,\n",
+            "dispatch_batches=None,\n",
+            "do_eval=False,\n",
+            "do_predict=False,\n",
+            "do_train=True,\n",
+            "eval_accumulation_steps=None,\n",
+            "eval_delay=0,\n",
+            "eval_steps=None,\n",
+            "evaluation_strategy=no,\n",
+            "fp16=False,\n",
+            "fp16_backend=auto,\n",
+            "fp16_full_eval=False,\n",
+            "fp16_opt_level=O1,\n",
+            "fsdp=[],\n",
+            "fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_grad_ckpt': False},\n",
+            "fsdp_min_num_params=0,\n",
+            "fsdp_transformer_layer_cls_to_wrap=None,\n",
+            "full_determinism=False,\n",
+            "gradient_accumulation_steps=1,\n",
+            "gradient_checkpointing=False,\n",
+            "gradient_checkpointing_kwargs=None,\n",
+            "greater_is_better=None,\n",
+            "group_by_length=False,\n",
+            "half_precision_backend=auto,\n",
+            "hub_always_push=False,\n",
+            "hub_model_id=None,\n",
+            "hub_private_repo=False,\n",
+            "hub_strategy=every_save,\n",
+            "hub_token=<HUB_TOKEN>,\n",
+            "ignore_data_skip=False,\n",
+            "include_inputs_for_metrics=False,\n",
+            "include_num_input_tokens_seen=False,\n",
+            "include_tokens_per_second=False,\n",
+            "jit_mode_eval=False,\n",
+            "label_names=None,\n",
+            "label_smoothing_factor=0.0,\n",
+            "learning_rate=5e-05,\n",
+            "length_column_name=length,\n",
+            "load_best_model_at_end=False,\n",
+            "local_rank=0,\n",
+            "log_level=passive,\n",
+            "log_level_replica=warning,\n",
+            "log_on_each_node=True,\n",
+            "logging_dir=./gpt2-drugscom_depression_reviews/runs/Dec09_12-33-34_8acb048be5fa,\n",
+            "logging_first_step=False,\n",
+            "logging_nan_inf_filter=True,\n",
+            "logging_steps=500,\n",
+            "logging_strategy=steps,\n",
+            "lr_scheduler_kwargs={},\n",
+            "lr_scheduler_type=linear,\n",
+            "max_grad_norm=1.0,\n",
+            "max_steps=-1,\n",
+            "metric_for_best_model=None,\n",
+            "mp_parameters=,\n",
+            "neftune_noise_alpha=None,\n",
+            "no_cuda=False,\n",
+            "num_train_epochs=5.0,\n",
+            "optim=adamw_torch,\n",
+            "optim_args=None,\n",
+            "output_dir=./gpt2-drugscom_depression_reviews,\n",
+            "overwrite_output_dir=True,\n",
+            "past_index=-1,\n",
+            "per_device_eval_batch_size=8,\n",
+            "per_device_train_batch_size=2,\n",
+            "prediction_loss_only=False,\n",
+            "push_to_hub=False,\n",
+            "push_to_hub_model_id=None,\n",
+            "push_to_hub_organization=None,\n",
+            "push_to_hub_token=<PUSH_TO_HUB_TOKEN>,\n",
+            "ray_scope=last,\n",
+            "remove_unused_columns=True,\n",
+            "report_to=['wandb'],\n",
+            "resume_from_checkpoint=None,\n",
+            "run_name=./gpt2-drugscom_depression_reviews,\n",
+            "save_on_each_node=False,\n",
+            "save_only_model=False,\n",
+            "save_safetensors=True,\n",
+            "save_steps=500,\n",
+            "save_strategy=steps,\n",
+            "save_total_limit=None,\n",
+            "seed=42,\n",
+            "skip_memory_metrics=True,\n",
+            "split_batches=False,\n",
+            "tf32=None,\n",
+            "torch_compile=False,\n",
+            "torch_compile_backend=None,\n",
+            "torch_compile_mode=None,\n",
+            "torchdynamo=None,\n",
+            "tpu_metrics_debug=False,\n",
+            "tpu_num_cores=None,\n",
+            "use_cpu=False,\n",
+            "use_ipex=False,\n",
+            "use_legacy_prediction_loop=False,\n",
+            "use_mps_device=False,\n",
+            "warmup_ratio=0.0,\n",
+            "warmup_steps=0,\n",
+            "weight_decay=0.0,\n",
+            ")\n",
+            "Using custom data configuration default-99d8365ea3008c01\n",
+            "12/09/2023 12:33:40 - INFO - datasets.builder - Using custom data configuration default-99d8365ea3008c01\n",
+            "Loading Dataset Infos from /usr/local/lib/python3.10/dist-packages/datasets/packaged_modules/text\n",
+            "12/09/2023 12:33:40 - INFO - datasets.info - Loading Dataset Infos from /usr/local/lib/python3.10/dist-packages/datasets/packaged_modules/text\n",
+            "Generating dataset text (/root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34)\n",
+            "12/09/2023 12:33:40 - INFO - datasets.builder - Generating dataset text (/root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34)\n",
+            "Downloading and preparing dataset text/default to /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34...\n",
+            "12/09/2023 12:33:40 - INFO - datasets.builder - Downloading and preparing dataset text/default to /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34...\n",
+            "Downloading data files: 100% 1/1 [00:00<00:00, 7256.58it/s]\n",
+            "Downloading took 0.0 min\n",
+            "12/09/2023 12:33:40 - INFO - datasets.download.download_manager - Downloading took 0.0 min\n",
+            "Checksum Computation took 0.0 min\n",
+            "12/09/2023 12:33:40 - INFO - datasets.download.download_manager - Checksum Computation took 0.0 min\n",
+            "Extracting data files: 100% 1/1 [00:00<00:00, 1198.72it/s]\n",
+            "Generating train split\n",
+            "12/09/2023 12:33:40 - INFO - datasets.builder - Generating train split\n",
+            "Generating train split: 12943 examples [00:00, 19902.65 examples/s]\n",
+            "Unable to verify splits sizes.\n",
+            "12/09/2023 12:33:41 - INFO - datasets.utils.info_utils - Unable to verify splits sizes.\n",
+            "Dataset text downloaded and prepared to /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34. Subsequent calls will reuse this data.\n",
+            "12/09/2023 12:33:41 - INFO - datasets.builder - Dataset text downloaded and prepared to /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34. Subsequent calls will reuse this data.\n",
+            "Using custom data configuration default-99d8365ea3008c01\n",
+            "12/09/2023 12:33:41 - INFO - datasets.builder - Using custom data configuration default-99d8365ea3008c01\n",
+            "Loading Dataset Infos from /usr/local/lib/python3.10/dist-packages/datasets/packaged_modules/text\n",
+            "12/09/2023 12:33:41 - INFO - datasets.info - Loading Dataset Infos from /usr/local/lib/python3.10/dist-packages/datasets/packaged_modules/text\n",
+            "Overwrite dataset info from restored data version if exists.\n",
+            "12/09/2023 12:33:41 - INFO - datasets.builder - Overwrite dataset info from restored data version if exists.\n",
+            "Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "12/09/2023 12:33:41 - INFO - datasets.info - Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "Found cached dataset text (/root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34)\n",
+            "12/09/2023 12:33:41 - INFO - datasets.builder - Found cached dataset text (/root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34)\n",
+            "Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "12/09/2023 12:33:41 - INFO - datasets.info - Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "Using custom data configuration default-99d8365ea3008c01\n",
+            "12/09/2023 12:33:41 - INFO - datasets.builder - Using custom data configuration default-99d8365ea3008c01\n",
+            "Loading Dataset Infos from /usr/local/lib/python3.10/dist-packages/datasets/packaged_modules/text\n",
+            "12/09/2023 12:33:41 - INFO - datasets.info - Loading Dataset Infos from /usr/local/lib/python3.10/dist-packages/datasets/packaged_modules/text\n",
+            "Overwrite dataset info from restored data version if exists.\n",
+            "12/09/2023 12:33:41 - INFO - datasets.builder - Overwrite dataset info from restored data version if exists.\n",
+            "Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "12/09/2023 12:33:41 - INFO - datasets.info - Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "Found cached dataset text (/root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34)\n",
+            "12/09/2023 12:33:41 - INFO - datasets.builder - Found cached dataset text (/root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34)\n",
+            "Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "12/09/2023 12:33:41 - INFO - datasets.info - Loading Dataset info from /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34\n",
+            "[INFO|configuration_utils.py:739] 2023-12-09 12:33:41,848 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/config.json\n",
+            "[INFO|configuration_utils.py:802] 2023-12-09 12:33:41,850 >> Model config GPT2Config {\n",
+            "  \"_name_or_path\": \"gpt2\",\n",
+            "  \"activation_function\": \"gelu_new\",\n",
+            "  \"architectures\": [\n",
+            "    \"GPT2LMHeadModel\"\n",
+            "  ],\n",
+            "  \"attn_pdrop\": 0.1,\n",
+            "  \"bos_token_id\": 50256,\n",
+            "  \"embd_pdrop\": 0.1,\n",
+            "  \"eos_token_id\": 50256,\n",
+            "  \"initializer_range\": 0.02,\n",
+            "  \"layer_norm_epsilon\": 1e-05,\n",
+            "  \"model_type\": \"gpt2\",\n",
+            "  \"n_ctx\": 1024,\n",
+            "  \"n_embd\": 768,\n",
+            "  \"n_head\": 12,\n",
+            "  \"n_inner\": null,\n",
+            "  \"n_layer\": 12,\n",
+            "  \"n_positions\": 1024,\n",
+            "  \"reorder_and_upcast_attn\": false,\n",
+            "  \"resid_pdrop\": 0.1,\n",
+            "  \"scale_attn_by_inverse_layer_idx\": false,\n",
+            "  \"scale_attn_weights\": true,\n",
+            "  \"summary_activation\": null,\n",
+            "  \"summary_first_dropout\": 0.1,\n",
+            "  \"summary_proj_to_labels\": true,\n",
+            "  \"summary_type\": \"cls_index\",\n",
+            "  \"summary_use_proj\": true,\n",
+            "  \"task_specific_params\": {\n",
+            "    \"text-generation\": {\n",
+            "      \"do_sample\": true,\n",
+            "      \"max_length\": 50\n",
+            "    }\n",
+            "  },\n",
+            "  \"transformers_version\": \"4.36.0.dev0\",\n",
+            "  \"use_cache\": true,\n",
+            "  \"vocab_size\": 50257\n",
+            "}\n",
+            "\n",
+            "[INFO|tokenization_auto.py:578] 2023-12-09 12:33:41,919 >> Could not locate the tokenizer configuration file, will try to use the model config instead.\n",
+            "[INFO|configuration_utils.py:739] 2023-12-09 12:33:41,991 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/config.json\n",
+            "[INFO|configuration_utils.py:802] 2023-12-09 12:33:41,992 >> Model config GPT2Config {\n",
+            "  \"_name_or_path\": \"gpt2\",\n",
+            "  \"activation_function\": \"gelu_new\",\n",
+            "  \"architectures\": [\n",
+            "    \"GPT2LMHeadModel\"\n",
+            "  ],\n",
+            "  \"attn_pdrop\": 0.1,\n",
+            "  \"bos_token_id\": 50256,\n",
+            "  \"embd_pdrop\": 0.1,\n",
+            "  \"eos_token_id\": 50256,\n",
+            "  \"initializer_range\": 0.02,\n",
+            "  \"layer_norm_epsilon\": 1e-05,\n",
+            "  \"model_type\": \"gpt2\",\n",
+            "  \"n_ctx\": 1024,\n",
+            "  \"n_embd\": 768,\n",
+            "  \"n_head\": 12,\n",
+            "  \"n_inner\": null,\n",
+            "  \"n_layer\": 12,\n",
+            "  \"n_positions\": 1024,\n",
+            "  \"reorder_and_upcast_attn\": false,\n",
+            "  \"resid_pdrop\": 0.1,\n",
+            "  \"scale_attn_by_inverse_layer_idx\": false,\n",
+            "  \"scale_attn_weights\": true,\n",
+            "  \"summary_activation\": null,\n",
+            "  \"summary_first_dropout\": 0.1,\n",
+            "  \"summary_proj_to_labels\": true,\n",
+            "  \"summary_type\": \"cls_index\",\n",
+            "  \"summary_use_proj\": true,\n",
+            "  \"task_specific_params\": {\n",
+            "    \"text-generation\": {\n",
+            "      \"do_sample\": true,\n",
+            "      \"max_length\": 50\n",
+            "    }\n",
+            "  },\n",
+            "  \"transformers_version\": \"4.36.0.dev0\",\n",
+            "  \"use_cache\": true,\n",
+            "  \"vocab_size\": 50257\n",
+            "}\n",
+            "\n",
+            "[INFO|tokenization_utils_base.py:2026] 2023-12-09 12:33:42,155 >> loading file vocab.json from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/vocab.json\n",
+            "[INFO|tokenization_utils_base.py:2026] 2023-12-09 12:33:42,155 >> loading file merges.txt from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/merges.txt\n",
+            "[INFO|tokenization_utils_base.py:2026] 2023-12-09 12:33:42,155 >> loading file tokenizer.json from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/tokenizer.json\n",
+            "[INFO|tokenization_utils_base.py:2026] 2023-12-09 12:33:42,155 >> loading file added_tokens.json from cache at None\n",
+            "[INFO|tokenization_utils_base.py:2026] 2023-12-09 12:33:42,155 >> loading file special_tokens_map.json from cache at None\n",
+            "[INFO|tokenization_utils_base.py:2026] 2023-12-09 12:33:42,155 >> loading file tokenizer_config.json from cache at None\n",
+            "[INFO|configuration_utils.py:739] 2023-12-09 12:33:42,156 >> loading configuration file config.json from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/config.json\n",
+            "[INFO|configuration_utils.py:802] 2023-12-09 12:33:42,157 >> Model config GPT2Config {\n",
+            "  \"_name_or_path\": \"gpt2\",\n",
+            "  \"activation_function\": \"gelu_new\",\n",
+            "  \"architectures\": [\n",
+            "    \"GPT2LMHeadModel\"\n",
+            "  ],\n",
+            "  \"attn_pdrop\": 0.1,\n",
+            "  \"bos_token_id\": 50256,\n",
+            "  \"embd_pdrop\": 0.1,\n",
+            "  \"eos_token_id\": 50256,\n",
+            "  \"initializer_range\": 0.02,\n",
+            "  \"layer_norm_epsilon\": 1e-05,\n",
+            "  \"model_type\": \"gpt2\",\n",
+            "  \"n_ctx\": 1024,\n",
+            "  \"n_embd\": 768,\n",
+            "  \"n_head\": 12,\n",
+            "  \"n_inner\": null,\n",
+            "  \"n_layer\": 12,\n",
+            "  \"n_positions\": 1024,\n",
+            "  \"reorder_and_upcast_attn\": false,\n",
+            "  \"resid_pdrop\": 0.1,\n",
+            "  \"scale_attn_by_inverse_layer_idx\": false,\n",
+            "  \"scale_attn_weights\": true,\n",
+            "  \"summary_activation\": null,\n",
+            "  \"summary_first_dropout\": 0.1,\n",
+            "  \"summary_proj_to_labels\": true,\n",
+            "  \"summary_type\": \"cls_index\",\n",
+            "  \"summary_use_proj\": true,\n",
+            "  \"task_specific_params\": {\n",
+            "    \"text-generation\": {\n",
+            "      \"do_sample\": true,\n",
+            "      \"max_length\": 50\n",
+            "    }\n",
+            "  },\n",
+            "  \"transformers_version\": \"4.36.0.dev0\",\n",
+            "  \"use_cache\": true,\n",
+            "  \"vocab_size\": 50257\n",
+            "}\n",
+            "\n",
+            "model.safetensors: 100% 548M/548M [00:06<00:00, 78.6MB/s]\n",
+            "[INFO|modeling_utils.py:3299] 2023-12-09 12:33:49,589 >> loading weights file model.safetensors from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/model.safetensors\n",
+            "[INFO|configuration_utils.py:826] 2023-12-09 12:33:49,639 >> Generate config GenerationConfig {\n",
+            "  \"bos_token_id\": 50256,\n",
+            "  \"eos_token_id\": 50256\n",
+            "}\n",
+            "\n",
+            "[INFO|modeling_utils.py:4139] 2023-12-09 12:33:51,494 >> All model checkpoint weights were used when initializing GPT2LMHeadModel.\n",
+            "\n",
+            "[INFO|modeling_utils.py:4147] 2023-12-09 12:33:51,494 >> All the weights of GPT2LMHeadModel were initialized from the model checkpoint at gpt2.\n",
+            "If your task is similar to the task the model of the checkpoint was trained on, you can already use GPT2LMHeadModel for predictions without further training.\n",
+            "generation_config.json: 100% 124/124 [00:00<00:00, 629kB/s]\n",
+            "[INFO|configuration_utils.py:781] 2023-12-09 12:33:51,630 >> loading configuration file generation_config.json from cache at /root/.cache/huggingface/hub/models--gpt2/snapshots/11c5a3d5811f50298f278a704980280950aedb10/generation_config.json\n",
+            "[INFO|configuration_utils.py:826] 2023-12-09 12:33:51,630 >> Generate config GenerationConfig {\n",
+            "  \"bos_token_id\": 50256,\n",
+            "  \"eos_token_id\": 50256\n",
+            "}\n",
+            "\n",
+            "Running tokenizer on dataset:   0% 0/12296 [00:00<?, ? examples/s]Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-75bdeafd5e91ab76.arrow\n",
+            "12/09/2023 12:33:52 - INFO - datasets.arrow_dataset - Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-75bdeafd5e91ab76.arrow\n",
+            "Running tokenizer on dataset: 100% 12296/12296 [00:10<00:00, 1160.82 examples/s]\n",
+            "Running tokenizer on dataset:   0% 0/647 [00:00<?, ? examples/s]Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-4b9b05e8867d8937.arrow\n",
+            "12/09/2023 12:34:02 - INFO - datasets.arrow_dataset - Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-4b9b05e8867d8937.arrow\n",
+            "Running tokenizer on dataset: 100% 647/647 [00:00<00:00, 1483.74 examples/s]\n",
+            "Grouping texts in chunks of 1024:   0% 0/12296 [00:00<?, ? examples/s]Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-7968eaff495c5a36.arrow\n",
+            "12/09/2023 12:34:03 - INFO - datasets.arrow_dataset - Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-7968eaff495c5a36.arrow\n",
+            "Grouping texts in chunks of 1024: 100% 12296/12296 [00:05<00:00, 2431.06 examples/s]\n",
+            "Grouping texts in chunks of 1024:   0% 0/647 [00:00<?, ? examples/s]Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-949bae4093020651.arrow\n",
+            "12/09/2023 12:34:08 - INFO - datasets.arrow_dataset - Caching processed dataset at /root/.cache/huggingface/datasets/text/default-99d8365ea3008c01/0.0.0/c4a140d10f020282918b5dd1b8a49f0104729c6177f60a6b49ec2a365ec69f34/cache-949bae4093020651.arrow\n",
+            "Grouping texts in chunks of 1024: 100% 647/647 [00:00<00:00, 1340.88 examples/s]\n",
+            "[INFO|trainer.py:1706] 2023-12-09 12:34:09,306 >> ***** Running training *****\n",
+            "[INFO|trainer.py:1707] 2023-12-09 12:34:09,306 >>   Num examples = 4,308\n",
+            "[INFO|trainer.py:1708] 2023-12-09 12:34:09,306 >>   Num Epochs = 5\n",
+            "[INFO|trainer.py:1709] 2023-12-09 12:34:09,306 >>   Instantaneous batch size per device = 2\n",
+            "[INFO|trainer.py:1712] 2023-12-09 12:34:09,307 >>   Total train batch size (w. parallel, distributed & accumulation) = 2\n",
+            "[INFO|trainer.py:1713] 2023-12-09 12:34:09,307 >>   Gradient Accumulation steps = 1\n",
+            "[INFO|trainer.py:1714] 2023-12-09 12:34:09,307 >>   Total optimization steps = 10,770\n",
+            "[INFO|trainer.py:1715] 2023-12-09 12:34:09,307 >>   Number of trainable parameters = 124,439,808\n",
+            "[INFO|integration_utils.py:722] 2023-12-09 12:34:09,308 >> Automatic Weights & Biases logging enabled, to disable set os.environ[\"WANDB_DISABLED\"] = \"true\"\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: Currently logged in as: \u001b[33mzakia-salod\u001b[0m (\u001b[33mteam-zakia\u001b[0m). Use \u001b[1m`wandb login --relogin`\u001b[0m to force relogin\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: Tracking run with wandb version 0.16.1\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: Run data is saved locally in \u001b[35m\u001b[1m/content/wandb/run-20231209_123409-bsxtbazc\u001b[0m\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: Run \u001b[1m`wandb offline`\u001b[0m to turn off syncing.\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: Syncing run \u001b[33mradiant-firebrand-8\u001b[0m\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: ⭐️ View project at \u001b[34m\u001b[4mhttps://wandb.ai/team-zakia/huggingface\u001b[0m\n",
+            "\u001b[34m\u001b[1mwandb\u001b[0m: 🚀 View run at \u001b[34m\u001b[4mhttps://wandb.ai/team-zakia/huggingface/runs/bsxtbazc\u001b[0m\n",
+            "{'loss': 0.7369, 'learning_rate': 4.767873723305478e-05, 'epoch': 0.23}\n",
+            "  5% 500/10770 [06:09<2:07:10,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 12:40:20,104 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 12:40:20,106 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 12:40:20,107 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 12:40:22,146 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 12:40:22,148 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 12:40:22,148 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-500/special_tokens_map.json\n",
+            "{'loss': 0.6937, 'learning_rate': 4.535747446610957e-05, 'epoch': 0.46}\n",
+            "  9% 1000/10770 [12:29<2:00:13,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 12:46:40,220 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 12:46:40,221 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 12:46:40,222 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 12:46:44,783 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 12:46:44,785 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 12:46:44,788 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1000/special_tokens_map.json\n",
+            "{'loss': 0.6743, 'learning_rate': 4.303621169916435e-05, 'epoch': 0.7}\n",
+            " 14% 1500/10770 [18:49<1:54:18,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 12:52:59,800 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 12:52:59,801 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 12:52:59,802 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 12:53:01,644 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 12:53:01,645 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 12:53:01,646 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-1500/special_tokens_map.json\n",
+            "{'loss': 0.6565, 'learning_rate': 4.071494893221913e-05, 'epoch': 0.93}\n",
+            " 19% 2000/10770 [25:05<1:48:39,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 12:59:15,817 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 12:59:15,818 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 12:59:15,819 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 12:59:17,871 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 12:59:17,872 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 12:59:17,873 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2000/special_tokens_map.json\n",
+            "{'loss': 0.6237, 'learning_rate': 3.8393686165273915e-05, 'epoch': 1.16}\n",
+            " 23% 2500/10770 [31:29<1:41:45,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:05:39,969 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:05:39,971 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:05:39,971 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:05:43,191 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:05:43,192 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:05:43,192 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-2500/special_tokens_map.json\n",
+            "{'loss': 0.6237, 'learning_rate': 3.607242339832869e-05, 'epoch': 1.39}\n",
+            " 28% 3000/10770 [37:48<1:35:53,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:11:58,874 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:11:58,876 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:11:58,877 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:12:00,785 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:12:00,787 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:12:00,788 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3000/special_tokens_map.json\n",
+            "{'loss': 0.6167, 'learning_rate': 3.375116063138347e-05, 'epoch': 1.62}\n",
+            " 32% 3500/10770 [44:04<1:29:41,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:18:15,187 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:18:15,188 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:18:15,189 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:18:19,719 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:18:19,720 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:18:19,721 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-3500/special_tokens_map.json\n",
+            "{'loss': 0.6143, 'learning_rate': 3.1429897864438255e-05, 'epoch': 1.86}\n",
+            " 37% 4000/10770 [50:25<1:23:55,  1.34it/s][INFO|trainer.py:2877] 2023-12-09 13:24:35,665 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:24:35,667 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:24:35,668 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:24:37,671 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:24:37,672 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:24:37,673 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4000/special_tokens_map.json\n",
+            "{'loss': 0.6003, 'learning_rate': 2.9108635097493035e-05, 'epoch': 2.09}\n",
+            " 42% 4500/10770 [56:41<1:17:15,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:30:52,074 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:30:52,076 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:30:52,079 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:30:54,225 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:30:54,227 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:30:54,228 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-4500/special_tokens_map.json\n",
+            "{'loss': 0.5719, 'learning_rate': 2.678737233054782e-05, 'epoch': 2.32}\n",
+            " 46% 5000/10770 [1:02:58<1:11:11,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:37:08,692 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:37:08,693 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:37:08,694 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:37:10,841 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:37:10,843 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:37:10,846 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5000/special_tokens_map.json\n",
+            "{'loss': 0.5824, 'learning_rate': 2.44661095636026e-05, 'epoch': 2.55}\n",
+            " 51% 5500/10770 [1:09:14<1:04:55,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:43:25,363 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:43:25,364 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:43:25,365 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:43:27,212 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:43:27,214 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:43:27,214 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-5500/special_tokens_map.json\n",
+            "{'loss': 0.5766, 'learning_rate': 2.214484679665738e-05, 'epoch': 2.79}\n",
+            " 56% 6000/10770 [1:15:31<58:47,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:49:41,984 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:49:41,986 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:49:41,987 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:49:43,886 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:49:43,887 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:49:43,888 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6000/special_tokens_map.json\n",
+            "{'loss': 0.5798, 'learning_rate': 1.9823584029712165e-05, 'epoch': 3.02}\n",
+            " 60% 6500/10770 [1:21:47<52:35,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 13:55:58,283 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 13:55:58,284 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 13:55:58,286 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 13:56:00,185 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 13:56:00,187 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 13:56:00,187 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-6500/special_tokens_map.json\n",
+            "{'loss': 0.5537, 'learning_rate': 1.7502321262766948e-05, 'epoch': 3.25}\n",
+            " 65% 7000/10770 [1:28:03<46:29,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 14:02:14,355 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:02:14,357 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:02:14,358 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:02:17,406 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:02:17,407 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:02:17,408 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7000/special_tokens_map.json\n",
+            "{'loss': 0.5553, 'learning_rate': 1.518105849582173e-05, 'epoch': 3.48}\n",
+            " 70% 7500/10770 [1:34:22<40:34,  1.34it/s][INFO|trainer.py:2877] 2023-12-09 14:08:33,229 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:08:33,231 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:08:33,232 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:08:35,154 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:08:35,155 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:08:35,155 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-7500/special_tokens_map.json\n",
+            "{'loss': 0.5479, 'learning_rate': 1.2859795728876508e-05, 'epoch': 3.71}\n",
+            " 74% 8000/10770 [1:40:38<34:27,  1.34it/s][INFO|trainer.py:2877] 2023-12-09 14:14:49,206 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:14:49,209 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:14:49,210 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:14:51,216 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:14:51,218 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:14:51,219 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8000/special_tokens_map.json\n",
+            "{'loss': 0.5538, 'learning_rate': 1.0538532961931291e-05, 'epoch': 3.95}\n",
+            " 79% 8500/10770 [1:46:55<27:57,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 14:21:06,112 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:21:06,114 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:21:06,115 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:21:08,009 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:21:08,011 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:21:08,012 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-8500/special_tokens_map.json\n",
+            "{'loss': 0.5429, 'learning_rate': 8.217270194986073e-06, 'epoch': 4.18}\n",
+            " 84% 9000/10770 [1:53:11<21:43,  1.36it/s][INFO|trainer.py:2877] 2023-12-09 14:27:22,134 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:27:22,135 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:27:22,136 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:27:24,023 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:27:24,024 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:27:24,025 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9000/special_tokens_map.json\n",
+            "{'loss': 0.5361, 'learning_rate': 5.896007428040855e-06, 'epoch': 4.41}\n",
+            " 88% 9500/10770 [1:59:27<15:39,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 14:33:38,093 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:33:38,095 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:33:38,096 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:33:39,987 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:33:39,988 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:33:39,988 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-9500/special_tokens_map.json\n",
+            "{'loss': 0.5425, 'learning_rate': 3.574744661095636e-06, 'epoch': 4.64}\n",
+            " 93% 10000/10770 [2:05:51<09:27,  1.36it/s][INFO|trainer.py:2877] 2023-12-09 14:40:02,357 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10000\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:40:02,358 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10000/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:40:02,359 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10000/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:40:08,494 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10000/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:40:08,501 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10000/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:40:08,502 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10000/special_tokens_map.json\n",
+            "{'loss': 0.5388, 'learning_rate': 1.253481894150418e-06, 'epoch': 4.87}\n",
+            " 97% 10500/10770 [2:12:12<03:20,  1.35it/s][INFO|trainer.py:2877] 2023-12-09 14:46:23,328 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10500\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:46:23,330 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10500/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:46:23,331 >> Configuration saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10500/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:46:25,338 >> Model weights saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10500/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:46:25,339 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10500/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:46:25,339 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/tmp-checkpoint-10500/special_tokens_map.json\n",
+            "100% 10770/10770 [2:15:39<00:00,  1.36it/s][INFO|trainer.py:1947] 2023-12-09 14:49:49,384 >> \n",
+            "\n",
+            "Training completed. Do not forget to share your model on huggingface.co/models =)\n",
+            "\n",
+            "\n",
+            "{'train_runtime': 8140.1135, 'train_samples_per_second': 2.646, 'train_steps_per_second': 1.323, 'train_loss': 0.5943892072502285, 'epoch': 5.0}\n",
+            "100% 10770/10770 [2:15:39<00:00,  1.32it/s]\n",
+            "[INFO|trainer.py:2877] 2023-12-09 14:49:49,424 >> Saving model checkpoint to ./gpt2-drugscom_depression_reviews\n",
+            "[INFO|configuration_utils.py:483] 2023-12-09 14:49:49,429 >> Configuration saved in ./gpt2-drugscom_depression_reviews/config.json\n",
+            "[INFO|configuration_utils.py:594] 2023-12-09 14:49:49,430 >> Configuration saved in ./gpt2-drugscom_depression_reviews/generation_config.json\n",
+            "[INFO|modeling_utils.py:2349] 2023-12-09 14:49:51,305 >> Model weights saved in ./gpt2-drugscom_depression_reviews/pytorch_model.bin\n",
+            "[INFO|tokenization_utils_base.py:2432] 2023-12-09 14:49:51,306 >> tokenizer config file saved in ./gpt2-drugscom_depression_reviews/tokenizer_config.json\n",
+            "[INFO|tokenization_utils_base.py:2441] 2023-12-09 14:49:51,307 >> Special tokens file saved in ./gpt2-drugscom_depression_reviews/special_tokens_map.json\n",
+            "***** train metrics *****\n",
+            "  epoch                    =        5.0\n",
+            "  train_loss               =     0.5944\n",
+            "  train_runtime            = 2:15:40.11\n",
+            "  train_samples            =       4308\n",
+            "  train_samples_per_second =      2.646\n",
+            "  train_steps_per_second   =      1.323\n",
+            "[INFO|modelcard.py:452] 2023-12-09 14:49:51,959 >> Dropping the following result as it does not have all the necessary fields:\n",
+            "{'task': {'name': 'Causal Language Modeling', 'type': 'text-generation'}}\n"
+          ]
+        }
+      ],
+      "source": [
+        "# Fine-tuning command for Google Colab\n",
+        "# Note: run_clm.py should be present in the working directory\n",
+        "!python run_clm.py \\\n",
+        "    --train_file train_dataset.txt \\\n",
+        "    --model_name_or_path gpt2 \\\n",
+        "    --output_dir ./gpt2-drugscom_depression_reviews \\\n",
+        "    --do_train \\\n",
+        "    --per_device_train_batch_size 2 \\\n",
+        "    --num_train_epochs 5 \\\n",
+        "    --overwrite_output_dir \\\n",
+        "    --use_fast_tokenizer True \\\n",
+        "    --report_to \"wandb\""
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {
+        "id": "EFH89vCQ0pMf"
+      },
+      "source": [
+        "## STEP 5: SAVE THE FINE-TUNED GPT-2 MODEL: gpt2-drugscom_depression_reviews"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "LthX-XqW5Agq"
+      },
+      "outputs": [],
+      "source": [
+        "from huggingface_hub import notebook_login # To log to our Hugging Face account to be able to upload models to the Hub."
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "X_g76SB35CDE",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 145,
+          "referenced_widgets": [
+            "9290ccab8e20428a8c6a61e2ce67e0c7",
+            "95db8459bbae45c9ac3b05898c06dc70",
+            "983e2f804ad24fe3b34a33c800417ca2",
+            "231839169648413991d11e48bb43a7a7",
+            "bd9f134683a84df09aadcab94fad8831",
+            "f6933a18f84f4678a360ffcc3e13f083",
+            "15fdfbc426924d5091533d4462c9b53b",
+            "44aa945dc5e743b8a6aa3f3078caac77",
+            "fa8d2f6085ab4a31b61270aa6a1ea4c5",
+            "7f3d379eb619439c98379c9b4f0795ec",
+            "6b048e6506aa454b9c5858828fbcd40c",
+            "4690ab39fdb348359cb42526a3ebeb86",
+            "0c3bc5d3a65d4a8980e321caf5a5d4f6",
+            "b5ea618a490448b4aa95637a03f9adb0",
+            "379537673a7b46a8bdcdbfdf346be8dd",
+            "bef797322e0940a0a4b0116bc502480d",
+            "8f9941f463cc4e7688daf9d70fb8937e",
+            "9502b7944c4948cd9850aa515ba5f2d3",
+            "56eeae6e18784dd2903addb05112e495",
+            "8bc3ad27022a45c29c0d7eac1da9956f",
+            "9fec786322d047299a289cb80487ed52",
+            "99801da567c5403284ac7555ea967dd3",
+            "d2dfcff538f54c449386cc843c2e7c6d",
+            "7728b2f00d96487a9397a55c3cd336ce",
+            "77dfd24b45f24da09dfe0c426d741aa7",
+            "c5313e81a1784396afd3cdc03f0920ea",
+            "65e11d956f2b4f59a9ccc26176cd4a58",
+            "452967ad3cd94a21bcde37e3ea2784eb",
+            "9fbe272599ed450e840a918a8b6a1fbc",
+            "d3c9c4c34afd4ebfb8b911c2aac756b5",
+            "1852d779e7c8448d87ce381d4348868d",
+            "6a2cbfb5e7b64c8cac850d0b05ce8f47"
+          ]
+        },
+        "outputId": "52df4679-af8f-4961-b006-6d11ee1ec163"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "VBox(children=(HTML(value='<center> <img\\nsrc=https://huggingface.co/front/assets/huggingface_logo-noborder.sv…"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "9290ccab8e20428a8c6a61e2ce67e0c7"
+            }
+          },
+          "metadata": {}
+        }
+      ],
+      "source": [
+        "# Login to Hugging Face within the notebook\n",
+        "notebook_login()\n",
+        "!git config --global credential.helper store"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "G1T1zFHM5F8m"
+      },
+      "outputs": [],
+      "source": [
+        "model_directory = \"./gpt2-drugscom_depression_reviews\""
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "NOV3BMvr5HLO"
+      },
+      "outputs": [],
+      "source": [
+        "# Load the fine-tuned model\n",
+        "model = GPT2LMHeadModel.from_pretrained(model_directory)"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {
+        "id": "qKa1fcoF3dc8",
+        "colab": {
+          "base_uri": "https://localhost:8080/",
+          "height": 84,
+          "referenced_widgets": [
+            "d2ce25900eae4b41b8107401f174b78d",
+            "84433208e40d4271b4db101602d267ef",
+            "c58122cb38ef4486a96966ac861cf913",
+            "121add4c3f014c04926701a1da4782f0",
+            "74c8e8e472904e6c9651509d8bfc6a50",
+            "9ada60f4d6374e3ab1d616970fea8b57",
+            "7ce21dd248a047fe8b4a8be37a05bfac",
+            "a3a9ece98352459baa56c051886a8924",
+            "ea47214f28704e6ea43cb064bf9f5294",
+            "c43afb5ad3ed429285c3a6d871a734d7",
+            "62a83b49f10f49ee9a8b85ad601b45c0"
+          ]
+        },
+        "outputId": "c4c5da4a-cd4f-4824-fedc-02471783f268"
+      },
+      "outputs": [
+        {
+          "output_type": "display_data",
+          "data": {
+            "text/plain": [
+              "model.safetensors:   0%|          | 0.00/498M [00:00<?, ?B/s]"
+            ],
+            "application/vnd.jupyter.widget-view+json": {
+              "version_major": 2,
+              "version_minor": 0,
+              "model_id": "d2ce25900eae4b41b8107401f174b78d"
+            }
+          },
+          "metadata": {}
+        },
+        {
+          "output_type": "execute_result",
+          "data": {
+            "text/plain": [
+              "CommitInfo(commit_url='https://huggingface.co/Zakia/gpt2-drugscom_depression_reviews/commit/1b4ea22d5bb925164a9a006031dd603f292807a0', commit_message='Upload tokenizer', commit_description='', oid='1b4ea22d5bb925164a9a006031dd603f292807a0', pr_url=None, pr_revision=None, pr_num=None)"
+            ]
+          },
+          "metadata": {},
+          "execution_count": 33
+        }
+      ],
+      "source": [
+        "# Save and push the model to the hub\n",
+        "model.push_to_hub(\"gpt2-drugscom_depression_reviews\")\n",
+        "tokenizer.push_to_hub(\"gpt2-drugscom_depression_reviews\")"
+      ]
+    }
+  ],
+  "metadata": {
+    "accelerator": "GPU",
+    "colab": {
+      "provenance": [],
+      "gpuType": "T4"
+    },
+    "kernelspec": {
+      "display_name": "Python 3",
+      "name": "python3"
+    },
+    "language_info": {
+      "name": "python"
+    },
+    "widgets": {
+      "application/vnd.jupyter.widget-state+json": {
+        "b45c8c73698048aeaee4d34b28d6a27a": {
+          "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_90e79e1e73ae489199ff3811dca9adae",
+              "IPY_MODEL_e59b83647a7f483da05b37429322c469",
+              "IPY_MODEL_66ec2664647846d8b4da85fee9e32df1"
+            ],
+            "layout": "IPY_MODEL_6b6eec8ea25445a6bc24b88778e1edbf"
+          }
+        },
+        "90e79e1e73ae489199ff3811dca9adae": {
+          "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_f2722d9ef4484a2a90ca9d6319d04345",
+            "placeholder": "​",
+            "style": "IPY_MODEL_9cc770f5d35e421cb07b44c9f7e2a730",
+            "value": "Downloading readme: 100%"
+          }
+        },
+        "e59b83647a7f483da05b37429322c469": {
+          "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_2c52eb43c73e4c7f8831bd6f71809570",
+            "max": 6718,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_02587cf05a5547c4ba874c34bf4cec4b",
+            "value": 6718
+          }
+        },
+        "66ec2664647846d8b4da85fee9e32df1": {
+          "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_0707b92bf1a84a9384beba55789ce6ba",
+            "placeholder": "​",
+            "style": "IPY_MODEL_57e880d55b0548a0b3a0f61ccdcc930f",
+            "value": " 6.72k/6.72k [00:00&lt;00:00, 228kB/s]"
+          }
+        },
+        "6b6eec8ea25445a6bc24b88778e1edbf": {
+          "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
+          }
+        },
+        "f2722d9ef4484a2a90ca9d6319d04345": {
+          "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
+          }
+        },
+        "9cc770f5d35e421cb07b44c9f7e2a730": {
+          "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": ""
+          }
+        },
+        "2c52eb43c73e4c7f8831bd6f71809570": {
+          "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
+          }
+        },
+        "02587cf05a5547c4ba874c34bf4cec4b": {
+          "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": ""
+          }
+        },
+        "0707b92bf1a84a9384beba55789ce6ba": {
+          "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
+          }
+        },
+        "57e880d55b0548a0b3a0f61ccdcc930f": {
+          "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": ""
+          }
+        },
+        "10311a17cdf645f69a3edd08ca2cad62": {
+          "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_a6000f68e93d4bf1a343629032dcef04",
+              "IPY_MODEL_ac869d78cfe146879a2494579c77277d",
+              "IPY_MODEL_2e35f836eaed419a8431e0c68e8d2cc1"
+            ],
+            "layout": "IPY_MODEL_a3d694bb06194b138a7abdbf199a1d81"
+          }
+        },
+        "a6000f68e93d4bf1a343629032dcef04": {
+          "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_5bdaf9e1d6da4b808b2a704b3c8d973b",
+            "placeholder": "​",
+            "style": "IPY_MODEL_2e5a51a0fb4c4249b55d8af4b0506eca",
+            "value": "Downloading data files: 100%"
+          }
+        },
+        "ac869d78cfe146879a2494579c77277d": {
+          "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_b4fcbf969be2427b92046a1f528db2a6",
+            "max": 2,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_43c31088bf444fd39578f2fe630dc46e",
+            "value": 2
+          }
+        },
+        "2e35f836eaed419a8431e0c68e8d2cc1": {
+          "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_3f7cb58407064ebaa1ec3943309c95c3",
+            "placeholder": "​",
+            "style": "IPY_MODEL_ce17ffc4033b4866b48d5e61467db6c3",
+            "value": " 2/2 [00:03&lt;00:00,  1.60s/it]"
+          }
+        },
+        "a3d694bb06194b138a7abdbf199a1d81": {
+          "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
+          }
+        },
+        "5bdaf9e1d6da4b808b2a704b3c8d973b": {
+          "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
+          }
+        },
+        "2e5a51a0fb4c4249b55d8af4b0506eca": {
+          "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": ""
+          }
+        },
+        "b4fcbf969be2427b92046a1f528db2a6": {
+          "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
+          }
+        },
+        "43c31088bf444fd39578f2fe630dc46e": {
+          "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": ""
+          }
+        },
+        "3f7cb58407064ebaa1ec3943309c95c3": {
+          "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
+          }
+        },
+        "ce17ffc4033b4866b48d5e61467db6c3": {
+          "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": ""
+          }
+        },
+        "0ffc9ba03f924bc0ad48fa82c2e711e7": {
+          "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_0abb262f82df4d728a8f05bf992a5ec3",
+              "IPY_MODEL_09e3afd0ba9e4b5298d741ce825fdbfb",
+              "IPY_MODEL_f069002ce14648c3b0269ae3e377879c"
+            ],
+            "layout": "IPY_MODEL_1b6b3d597a024cfb9a15573cea12891e"
+          }
+        },
+        "0abb262f82df4d728a8f05bf992a5ec3": {
+          "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_e48c988e59b8462babda6b5173a337a6",
+            "placeholder": "​",
+            "style": "IPY_MODEL_43f34a6373a7449a83d86705351d4957",
+            "value": "Downloading data: 100%"
+          }
+        },
+        "09e3afd0ba9e4b5298d741ce825fdbfb": {
+          "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_3794962a26714723a2b7d345336db7ec",
+            "max": 84289175,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_19ee2670c44b46e2906b21a73b27ba88",
+            "value": 84289175
+          }
+        },
+        "f069002ce14648c3b0269ae3e377879c": {
+          "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_16a81359fc4a40dbb83d185c1db81ebf",
+            "placeholder": "​",
+            "style": "IPY_MODEL_8970468c315b413abb6541205fc3174d",
+            "value": " 84.3M/84.3M [00:02&lt;00:00, 37.5MB/s]"
+          }
+        },
+        "1b6b3d597a024cfb9a15573cea12891e": {
+          "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
+          }
+        },
+        "e48c988e59b8462babda6b5173a337a6": {
+          "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
+          }
+        },
+        "43f34a6373a7449a83d86705351d4957": {
+          "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": ""
+          }
+        },
+        "3794962a26714723a2b7d345336db7ec": {
+          "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
+          }
+        },
+        "19ee2670c44b46e2906b21a73b27ba88": {
+          "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": ""
+          }
+        },
+        "16a81359fc4a40dbb83d185c1db81ebf": {
+          "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
+          }
+        },
+        "8970468c315b413abb6541205fc3174d": {
+          "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": ""
+          }
+        },
+        "84ba5a9fb669445d88a8f18c86986a65": {
+          "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_74e6479c15a843c196b26eae181f3b50",
+              "IPY_MODEL_24f3bb13e552446ca3dbf1d5acdd306e",
+              "IPY_MODEL_adf7780fbbed4baf90f08b113553e0a3"
+            ],
+            "layout": "IPY_MODEL_59b48ccd44734f2fb869291b1c52a56b"
+          }
+        },
+        "74e6479c15a843c196b26eae181f3b50": {
+          "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_bff2b32d8b624af98814c950bf2934d9",
+            "placeholder": "​",
+            "style": "IPY_MODEL_8b5cf7a76357475d8156886940ad2790",
+            "value": "Downloading data: 100%"
+          }
+        },
+        "24f3bb13e552446ca3dbf1d5acdd306e": {
+          "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_a8a19e8b09a84ea9a29a6dd8312d2110",
+            "max": 28071166,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_654b39a5572f42bfa93ba73c4b251fbd",
+            "value": 28071166
+          }
+        },
+        "adf7780fbbed4baf90f08b113553e0a3": {
+          "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_76464bf094c74cd68f3fb5e9975ff56e",
+            "placeholder": "​",
+            "style": "IPY_MODEL_1a3a890317814810834c117ffe62f390",
+            "value": " 28.1M/28.1M [00:00&lt;00:00, 38.4MB/s]"
+          }
+        },
+        "59b48ccd44734f2fb869291b1c52a56b": {
+          "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
+          }
+        },
+        "bff2b32d8b624af98814c950bf2934d9": {
+          "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
+          }
+        },
+        "8b5cf7a76357475d8156886940ad2790": {
+          "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": ""
+          }
+        },
+        "a8a19e8b09a84ea9a29a6dd8312d2110": {
+          "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
+          }
+        },
+        "654b39a5572f42bfa93ba73c4b251fbd": {
+          "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": ""
+          }
+        },
+        "76464bf094c74cd68f3fb5e9975ff56e": {
+          "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
+          }
+        },
+        "1a3a890317814810834c117ffe62f390": {
+          "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": ""
+          }
+        },
+        "a6687ff1f33a42988536dc04ccf824ad": {
+          "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_6e79a05a0f6f4e4c8b0ce54d47180df1",
+              "IPY_MODEL_c7f71309b9a14d9b85a9953f1c5ef919",
+              "IPY_MODEL_ad491e5e0a8d4106b6942f20206081f4"
+            ],
+            "layout": "IPY_MODEL_a08bd346d672441994b14340f57fa298"
+          }
+        },
+        "6e79a05a0f6f4e4c8b0ce54d47180df1": {
+          "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_d9166639b16c423fa902ef9a23a6582f",
+            "placeholder": "​",
+            "style": "IPY_MODEL_0e564d1c6ae8447fb431032ee5d64199",
+            "value": "Extracting data files: 100%"
+          }
+        },
+        "c7f71309b9a14d9b85a9953f1c5ef919": {
+          "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_c22ad507940c46d0bf34dc5efe8ffee3",
+            "max": 2,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_38331b41172b47fa96b2b7950dd4f94b",
+            "value": 2
+          }
+        },
+        "ad491e5e0a8d4106b6942f20206081f4": {
+          "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_15038217fa3c4c43990aeb222792feed",
+            "placeholder": "​",
+            "style": "IPY_MODEL_8bad5a126935481497fdb16ad2dacbb5",
+            "value": " 2/2 [00:00&lt;00:00, 101.73it/s]"
+          }
+        },
+        "a08bd346d672441994b14340f57fa298": {
+          "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
+          }
+        },
+        "d9166639b16c423fa902ef9a23a6582f": {
+          "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
+          }
+        },
+        "0e564d1c6ae8447fb431032ee5d64199": {
+          "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": ""
+          }
+        },
+        "c22ad507940c46d0bf34dc5efe8ffee3": {
+          "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
+          }
+        },
+        "38331b41172b47fa96b2b7950dd4f94b": {
+          "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": ""
+          }
+        },
+        "15038217fa3c4c43990aeb222792feed": {
+          "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
+          }
+        },
+        "8bad5a126935481497fdb16ad2dacbb5": {
+          "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": ""
+          }
+        },
+        "28b5d0115eb44b9483c9f7d345d644e1": {
+          "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_5aa2e3b0f8024725a260898904739acb",
+              "IPY_MODEL_bdb27820a5204529987b9e2df414df78",
+              "IPY_MODEL_f4f30d7e47864e4c9515ef6195661782"
+            ],
+            "layout": "IPY_MODEL_c4230c8b74d540afa5d2e885310dbf6b"
+          }
+        },
+        "5aa2e3b0f8024725a260898904739acb": {
+          "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_bf9000b17de04dd1bd522f94a3eb1ed9",
+            "placeholder": "​",
+            "style": "IPY_MODEL_97411c1fec18481cbd5bcab1e794cc7f",
+            "value": "Generating train split: "
+          }
+        },
+        "bdb27820a5204529987b9e2df414df78": {
+          "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_875e1cbc8e634cbaa00a22ff3c9d62f8",
+            "max": 1,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_52d6693766714e91b126d9180dd03346",
+            "value": 1
+          }
+        },
+        "f4f30d7e47864e4c9515ef6195661782": {
+          "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_f247e069bb4147e0b6019d66d851ede9",
+            "placeholder": "​",
+            "style": "IPY_MODEL_6a509c81bfff4f30bc7d2de029bdc93c",
+            "value": " 161297/0 [00:01&lt;00:00, 137488.63 examples/s]"
+          }
+        },
+        "c4230c8b74d540afa5d2e885310dbf6b": {
+          "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
+          }
+        },
+        "bf9000b17de04dd1bd522f94a3eb1ed9": {
+          "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
+          }
+        },
+        "97411c1fec18481cbd5bcab1e794cc7f": {
+          "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": ""
+          }
+        },
+        "875e1cbc8e634cbaa00a22ff3c9d62f8": {
+          "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"
+          }
+        },
+        "52d6693766714e91b126d9180dd03346": {
+          "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": ""
+          }
+        },
+        "f247e069bb4147e0b6019d66d851ede9": {
+          "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
+          }
+        },
+        "6a509c81bfff4f30bc7d2de029bdc93c": {
+          "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": ""
+          }
+        },
+        "36a765c57fc349eab616355610980324": {
+          "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_210adeaaf64e40db8323e085794e7d7a",
+              "IPY_MODEL_b8b1a3dfbb984ba08b1c862b26af8dbe",
+              "IPY_MODEL_304df8debe3f4662a4171221d669798e"
+            ],
+            "layout": "IPY_MODEL_266e9676bce140b69ad73456851ff499"
+          }
+        },
+        "210adeaaf64e40db8323e085794e7d7a": {
+          "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_9b0b095167de49f4bc5efb5502b3d41c",
+            "placeholder": "​",
+            "style": "IPY_MODEL_af16b06528824cc9b0ce63fc4b4738ba",
+            "value": "Generating test split: "
+          }
+        },
+        "b8b1a3dfbb984ba08b1c862b26af8dbe": {
+          "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_daa61ca2c88c49be9a115fb81f01caac",
+            "max": 1,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_5edea6130ae64675a5aecbbaffcc5821",
+            "value": 1
+          }
+        },
+        "304df8debe3f4662a4171221d669798e": {
+          "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_1b55987264384ee3bd90ba9e97842b61",
+            "placeholder": "​",
+            "style": "IPY_MODEL_2a4ea7765a984268a58766f089744cdb",
+            "value": " 53766/0 [00:00&lt;00:00, 140438.31 examples/s]"
+          }
+        },
+        "266e9676bce140b69ad73456851ff499": {
+          "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
+          }
+        },
+        "9b0b095167de49f4bc5efb5502b3d41c": {
+          "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
+          }
+        },
+        "af16b06528824cc9b0ce63fc4b4738ba": {
+          "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": ""
+          }
+        },
+        "daa61ca2c88c49be9a115fb81f01caac": {
+          "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"
+          }
+        },
+        "5edea6130ae64675a5aecbbaffcc5821": {
+          "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": ""
+          }
+        },
+        "1b55987264384ee3bd90ba9e97842b61": {
+          "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
+          }
+        },
+        "2a4ea7765a984268a58766f089744cdb": {
+          "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": ""
+          }
+        },
+        "35b794e3143a43b9867050806d00d7c2": {
+          "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_c0aabbfff4ca48609969258d77daf36c",
+              "IPY_MODEL_a656a7cec90141f480a2d447e271edd4",
+              "IPY_MODEL_fbd866b694004b478d12d7c8d621c563"
+            ],
+            "layout": "IPY_MODEL_a67a7144c6674c8eb119e5898b2d0490"
+          }
+        },
+        "c0aabbfff4ca48609969258d77daf36c": {
+          "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_4fe2cf6e63824e74b008998b6378bf01",
+            "placeholder": "​",
+            "style": "IPY_MODEL_29c67f208b3f4b90a38bba90aa369dcd",
+            "value": "Filter: 100%"
+          }
+        },
+        "a656a7cec90141f480a2d447e271edd4": {
+          "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_db9cdd8b18f24795ba9bd91a875ce448",
+            "max": 161297,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_5abc8caabb924c958c213348fde64848",
+            "value": 161297
+          }
+        },
+        "fbd866b694004b478d12d7c8d621c563": {
+          "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_afe1ff9c1d014590bba254ee2285c5c4",
+            "placeholder": "​",
+            "style": "IPY_MODEL_0c4015e83ef142bfb41abfde20a3fa05",
+            "value": " 161297/161297 [00:01&lt;00:00, 116711.44 examples/s]"
+          }
+        },
+        "a67a7144c6674c8eb119e5898b2d0490": {
+          "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
+          }
+        },
+        "4fe2cf6e63824e74b008998b6378bf01": {
+          "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
+          }
+        },
+        "29c67f208b3f4b90a38bba90aa369dcd": {
+          "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": ""
+          }
+        },
+        "db9cdd8b18f24795ba9bd91a875ce448": {
+          "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
+          }
+        },
+        "5abc8caabb924c958c213348fde64848": {
+          "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": ""
+          }
+        },
+        "afe1ff9c1d014590bba254ee2285c5c4": {
+          "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
+          }
+        },
+        "0c4015e83ef142bfb41abfde20a3fa05": {
+          "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": ""
+          }
+        },
+        "361e2c414ad942f5b8c92a5dcd3b1774": {
+          "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_dd3dbc5532264fb7881f11b0f3554b38",
+              "IPY_MODEL_48f3c24e3f384907a0e359b92e952914",
+              "IPY_MODEL_caeba295e0a84b21b6bece98ab886e1f"
+            ],
+            "layout": "IPY_MODEL_0651146886cc4d969d9f9a2ed6d18c8a"
+          }
+        },
+        "dd3dbc5532264fb7881f11b0f3554b38": {
+          "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_18453188685544d69ffc475599cb57db",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5cdaa0d7536b42cbaa1025dcdf4c911f",
+            "value": "Filter: 100%"
+          }
+        },
+        "48f3c24e3f384907a0e359b92e952914": {
+          "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_1193555c580545759bfcbf6a1913a5d5",
+            "max": 160398,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_77ccf09db8ad4452ac0b307081b68b05",
+            "value": 160398
+          }
+        },
+        "caeba295e0a84b21b6bece98ab886e1f": {
+          "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_f859e097073941d5b6baf569a731f396",
+            "placeholder": "​",
+            "style": "IPY_MODEL_cb19c04f36f84e5cb1d96dc520c6566d",
+            "value": " 160398/160398 [00:04&lt;00:00, 41071.73 examples/s]"
+          }
+        },
+        "0651146886cc4d969d9f9a2ed6d18c8a": {
+          "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
+          }
+        },
+        "18453188685544d69ffc475599cb57db": {
+          "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
+          }
+        },
+        "5cdaa0d7536b42cbaa1025dcdf4c911f": {
+          "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": ""
+          }
+        },
+        "1193555c580545759bfcbf6a1913a5d5": {
+          "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
+          }
+        },
+        "77ccf09db8ad4452ac0b307081b68b05": {
+          "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": ""
+          }
+        },
+        "f859e097073941d5b6baf569a731f396": {
+          "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
+          }
+        },
+        "cb19c04f36f84e5cb1d96dc520c6566d": {
+          "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": ""
+          }
+        },
+        "48f3b3e586f74174a0232079ec543e2c": {
+          "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_57403d31cdb64f11add87566d4f5a325",
+              "IPY_MODEL_f9e6f05aeb904395a5d5a8a3ec3b9756",
+              "IPY_MODEL_5b787cdd23f44cb8805e055ccffdfe40"
+            ],
+            "layout": "IPY_MODEL_e00ea9a3b8cc433a8dffc1cf9d99f609"
+          }
+        },
+        "57403d31cdb64f11add87566d4f5a325": {
+          "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_c6cae5d28bd9480eb1eb5642132d4dd5",
+            "placeholder": "​",
+            "style": "IPY_MODEL_90e73a35476f4e74aba7c02f329a3536",
+            "value": "Map: 100%"
+          }
+        },
+        "f9e6f05aeb904395a5d5a8a3ec3b9756": {
+          "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_dea3da72e6d946029a3760a594356701",
+            "max": 9069,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_5c7f4c3d4ebc47a6ba27aeed52a89add",
+            "value": 9069
+          }
+        },
+        "5b787cdd23f44cb8805e055ccffdfe40": {
+          "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_23396629e70847a28c247157d27587e7",
+            "placeholder": "​",
+            "style": "IPY_MODEL_2262d8d8318e453a92e4ae9b37fa484c",
+            "value": " 9069/9069 [00:00&lt;00:00, 15922.48 examples/s]"
+          }
+        },
+        "e00ea9a3b8cc433a8dffc1cf9d99f609": {
+          "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
+          }
+        },
+        "c6cae5d28bd9480eb1eb5642132d4dd5": {
+          "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
+          }
+        },
+        "90e73a35476f4e74aba7c02f329a3536": {
+          "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": ""
+          }
+        },
+        "dea3da72e6d946029a3760a594356701": {
+          "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
+          }
+        },
+        "5c7f4c3d4ebc47a6ba27aeed52a89add": {
+          "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": ""
+          }
+        },
+        "23396629e70847a28c247157d27587e7": {
+          "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
+          }
+        },
+        "2262d8d8318e453a92e4ae9b37fa484c": {
+          "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": ""
+          }
+        },
+        "97478b946a2243ae8fb5e2dbc939fecd": {
+          "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_b468cc474d964795aa92e609d8f49c14",
+              "IPY_MODEL_c208ce8f7548438a886309c3a1d89a46",
+              "IPY_MODEL_3f11e4cb4ea2467685187f86c6ad1259"
+            ],
+            "layout": "IPY_MODEL_78db2d61bf444938bd55d74ee2962bc7"
+          }
+        },
+        "b468cc474d964795aa92e609d8f49c14": {
+          "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_1c31184a73fa4ded96c3fc98aa98e4dd",
+            "placeholder": "​",
+            "style": "IPY_MODEL_feb1dec1d9684a8ebc23a97372cd836e",
+            "value": "vocab.json: 100%"
+          }
+        },
+        "c208ce8f7548438a886309c3a1d89a46": {
+          "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_6a1f524d208041bc9f632658154e9604",
+            "max": 1042301,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_aa0834b1a9774b039c81c30abd225c75",
+            "value": 1042301
+          }
+        },
+        "3f11e4cb4ea2467685187f86c6ad1259": {
+          "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_e6bf66b3ae9246af8dfb901f44114036",
+            "placeholder": "​",
+            "style": "IPY_MODEL_a3606f6637814cadaddeec5b809a2031",
+            "value": " 1.04M/1.04M [00:00&lt;00:00, 9.03MB/s]"
+          }
+        },
+        "78db2d61bf444938bd55d74ee2962bc7": {
+          "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
+          }
+        },
+        "1c31184a73fa4ded96c3fc98aa98e4dd": {
+          "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
+          }
+        },
+        "feb1dec1d9684a8ebc23a97372cd836e": {
+          "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": ""
+          }
+        },
+        "6a1f524d208041bc9f632658154e9604": {
+          "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
+          }
+        },
+        "aa0834b1a9774b039c81c30abd225c75": {
+          "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": ""
+          }
+        },
+        "e6bf66b3ae9246af8dfb901f44114036": {
+          "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
+          }
+        },
+        "a3606f6637814cadaddeec5b809a2031": {
+          "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": ""
+          }
+        },
+        "c14073600faa4276baabf68a8463d203": {
+          "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_86eeb0998e1f4d79bfe1d82cb7dec973",
+              "IPY_MODEL_5d65c6687d87417193211910daf341c9",
+              "IPY_MODEL_6cf38e7be70f4297a4a4a76afb00f5eb"
+            ],
+            "layout": "IPY_MODEL_f3530c618bed4888a151573bcec3df84"
+          }
+        },
+        "86eeb0998e1f4d79bfe1d82cb7dec973": {
+          "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_d28f426d69a94954923823dbf6bca790",
+            "placeholder": "​",
+            "style": "IPY_MODEL_b4cc09eaece64d81a60cd88a4dfd8327",
+            "value": "merges.txt: 100%"
+          }
+        },
+        "5d65c6687d87417193211910daf341c9": {
+          "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_1194b9e87bf24c8190a7114527f153cb",
+            "max": 456318,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_2b42c838b36445c28ad2c1d5802b94a8",
+            "value": 456318
+          }
+        },
+        "6cf38e7be70f4297a4a4a76afb00f5eb": {
+          "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_37453894d05741b5b76fbb25e00f3f4f",
+            "placeholder": "​",
+            "style": "IPY_MODEL_4e19eea754254320b0c64b7518494ca8",
+            "value": " 456k/456k [00:00&lt;00:00, 29.6MB/s]"
+          }
+        },
+        "f3530c618bed4888a151573bcec3df84": {
+          "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
+          }
+        },
+        "d28f426d69a94954923823dbf6bca790": {
+          "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
+          }
+        },
+        "b4cc09eaece64d81a60cd88a4dfd8327": {
+          "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": ""
+          }
+        },
+        "1194b9e87bf24c8190a7114527f153cb": {
+          "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
+          }
+        },
+        "2b42c838b36445c28ad2c1d5802b94a8": {
+          "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": ""
+          }
+        },
+        "37453894d05741b5b76fbb25e00f3f4f": {
+          "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
+          }
+        },
+        "4e19eea754254320b0c64b7518494ca8": {
+          "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": ""
+          }
+        },
+        "f3be5e77bcc14067a0b8333488b6989a": {
+          "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_063a528402a041c58ca72450e8f94a47",
+              "IPY_MODEL_f2b64c328fee41b5b137fd97fca75a63",
+              "IPY_MODEL_fa00fe6d3f074db9965be53e4694a830"
+            ],
+            "layout": "IPY_MODEL_fbe2c79add1e4125b9e8bc1b24e2bc7f"
+          }
+        },
+        "063a528402a041c58ca72450e8f94a47": {
+          "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_a6e378877e2146948f3e412bd31252ed",
+            "placeholder": "​",
+            "style": "IPY_MODEL_681923cbf6874204898918586d4d15e5",
+            "value": "tokenizer.json: 100%"
+          }
+        },
+        "f2b64c328fee41b5b137fd97fca75a63": {
+          "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_5f9839dab22a475f95c0547fc5035a67",
+            "max": 1355256,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_14a56ddc2f654e31b599bfbfc967d879",
+            "value": 1355256
+          }
+        },
+        "fa00fe6d3f074db9965be53e4694a830": {
+          "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_7d3128fda8c844b1bc84fadd2997a09a",
+            "placeholder": "​",
+            "style": "IPY_MODEL_da3bc0b8f8644bd2834089beb5aff4d7",
+            "value": " 1.36M/1.36M [00:00&lt;00:00, 45.5MB/s]"
+          }
+        },
+        "fbe2c79add1e4125b9e8bc1b24e2bc7f": {
+          "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
+          }
+        },
+        "a6e378877e2146948f3e412bd31252ed": {
+          "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
+          }
+        },
+        "681923cbf6874204898918586d4d15e5": {
+          "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": ""
+          }
+        },
+        "5f9839dab22a475f95c0547fc5035a67": {
+          "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
+          }
+        },
+        "14a56ddc2f654e31b599bfbfc967d879": {
+          "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": ""
+          }
+        },
+        "7d3128fda8c844b1bc84fadd2997a09a": {
+          "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
+          }
+        },
+        "da3bc0b8f8644bd2834089beb5aff4d7": {
+          "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": ""
+          }
+        },
+        "80eba5b6545948269049d0b804fc8f40": {
+          "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_ac32cb2229e541bf8aef1f7dd2ada74b",
+              "IPY_MODEL_a796211c642c4840a2822833d00f3ac0",
+              "IPY_MODEL_cc3831986c9a41099813df38119f8605"
+            ],
+            "layout": "IPY_MODEL_faf882727a9e442ca422ee63d02357c9"
+          }
+        },
+        "ac32cb2229e541bf8aef1f7dd2ada74b": {
+          "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_88ecab3b211e4cf69a76bb7b83ae0f7f",
+            "placeholder": "​",
+            "style": "IPY_MODEL_2294c3691218465aba561ce60038a691",
+            "value": "config.json: 100%"
+          }
+        },
+        "a796211c642c4840a2822833d00f3ac0": {
+          "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_057e01cd640f4b6bb05dbce84b40002d",
+            "max": 665,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_0733564e11614f3a9e4480222a58286d",
+            "value": 665
+          }
+        },
+        "cc3831986c9a41099813df38119f8605": {
+          "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_6f8671899ef74ea48a78df36ace76266",
+            "placeholder": "​",
+            "style": "IPY_MODEL_21f9727629834203afd92e6896f22071",
+            "value": " 665/665 [00:00&lt;00:00, 36.2kB/s]"
+          }
+        },
+        "faf882727a9e442ca422ee63d02357c9": {
+          "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
+          }
+        },
+        "88ecab3b211e4cf69a76bb7b83ae0f7f": {
+          "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
+          }
+        },
+        "2294c3691218465aba561ce60038a691": {
+          "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": ""
+          }
+        },
+        "057e01cd640f4b6bb05dbce84b40002d": {
+          "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
+          }
+        },
+        "0733564e11614f3a9e4480222a58286d": {
+          "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": ""
+          }
+        },
+        "6f8671899ef74ea48a78df36ace76266": {
+          "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
+          }
+        },
+        "21f9727629834203afd92e6896f22071": {
+          "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": ""
+          }
+        },
+        "558e52210904446684aa3f3a52d03ada": {
+          "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_9ddcd9da2ef34296829adce85e4f9c97",
+              "IPY_MODEL_ddcf5f257a214b089f84ab0ed1320d53",
+              "IPY_MODEL_22c72cc5f7104b2e82cb0363eed14243"
+            ],
+            "layout": "IPY_MODEL_153c517831cc4a9099bdfc1e3723ae02"
+          }
+        },
+        "9ddcd9da2ef34296829adce85e4f9c97": {
+          "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_ef41fb1aee914dfe83a10191ff488ac4",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5e8d743b4c054f6b9fff9440522e9635",
+            "value": "Map: 100%"
+          }
+        },
+        "ddcf5f257a214b089f84ab0ed1320d53": {
+          "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_8199c19bba6944cd95276a43b7edfa8a",
+            "max": 9069,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_c46c0c2430ae4ea2a982425921282e13",
+            "value": 9069
+          }
+        },
+        "22c72cc5f7104b2e82cb0363eed14243": {
+          "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_537cb348212a4258af62b2cafdae8266",
+            "placeholder": "​",
+            "style": "IPY_MODEL_5e001995d8984ccd87b73d2d8a6c89ef",
+            "value": " 9069/9069 [00:08&lt;00:00, 1312.66 examples/s]"
+          }
+        },
+        "153c517831cc4a9099bdfc1e3723ae02": {
+          "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
+          }
+        },
+        "ef41fb1aee914dfe83a10191ff488ac4": {
+          "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
+          }
+        },
+        "5e8d743b4c054f6b9fff9440522e9635": {
+          "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": ""
+          }
+        },
+        "8199c19bba6944cd95276a43b7edfa8a": {
+          "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
+          }
+        },
+        "c46c0c2430ae4ea2a982425921282e13": {
+          "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": ""
+          }
+        },
+        "537cb348212a4258af62b2cafdae8266": {
+          "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
+          }
+        },
+        "5e001995d8984ccd87b73d2d8a6c89ef": {
+          "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": ""
+          }
+        },
+        "9290ccab8e20428a8c6a61e2ce67e0c7": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "VBoxModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "VBoxModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "VBoxView",
+            "box_style": "",
+            "children": [
+              "IPY_MODEL_9fec786322d047299a289cb80487ed52",
+              "IPY_MODEL_99801da567c5403284ac7555ea967dd3",
+              "IPY_MODEL_d2dfcff538f54c449386cc843c2e7c6d",
+              "IPY_MODEL_7728b2f00d96487a9397a55c3cd336ce"
+            ],
+            "layout": "IPY_MODEL_15fdfbc426924d5091533d4462c9b53b"
+          }
+        },
+        "95db8459bbae45c9ac3b05898c06dc70": {
+          "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_44aa945dc5e743b8a6aa3f3078caac77",
+            "placeholder": "​",
+            "style": "IPY_MODEL_fa8d2f6085ab4a31b61270aa6a1ea4c5",
+            "value": "<center> <img\nsrc=https://huggingface.co/front/assets/huggingface_logo-noborder.svg\nalt='Hugging Face'> <br> Copy a token from <a\nhref=\"https://huggingface.co/settings/tokens\" target=\"_blank\">your Hugging Face\ntokens page</a> and paste it below. <br> Immediately click login after copying\nyour token or it might be stored in plain text in this notebook file. </center>"
+          }
+        },
+        "983e2f804ad24fe3b34a33c800417ca2": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "PasswordModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "PasswordModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "PasswordView",
+            "continuous_update": true,
+            "description": "Token:",
+            "description_tooltip": null,
+            "disabled": false,
+            "layout": "IPY_MODEL_7f3d379eb619439c98379c9b4f0795ec",
+            "placeholder": "​",
+            "style": "IPY_MODEL_6b048e6506aa454b9c5858828fbcd40c",
+            "value": ""
+          }
+        },
+        "231839169648413991d11e48bb43a7a7": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "CheckboxModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "CheckboxModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "CheckboxView",
+            "description": "Add token as git credential?",
+            "description_tooltip": null,
+            "disabled": false,
+            "indent": true,
+            "layout": "IPY_MODEL_4690ab39fdb348359cb42526a3ebeb86",
+            "style": "IPY_MODEL_0c3bc5d3a65d4a8980e321caf5a5d4f6",
+            "value": true
+          }
+        },
+        "bd9f134683a84df09aadcab94fad8831": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ButtonModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "ButtonModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "ButtonView",
+            "button_style": "",
+            "description": "Login",
+            "disabled": false,
+            "icon": "",
+            "layout": "IPY_MODEL_b5ea618a490448b4aa95637a03f9adb0",
+            "style": "IPY_MODEL_379537673a7b46a8bdcdbfdf346be8dd",
+            "tooltip": ""
+          }
+        },
+        "f6933a18f84f4678a360ffcc3e13f083": {
+          "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_bef797322e0940a0a4b0116bc502480d",
+            "placeholder": "​",
+            "style": "IPY_MODEL_8f9941f463cc4e7688daf9d70fb8937e",
+            "value": "\n<b>Pro Tip:</b> If you don't already have one, you can create a dedicated\n'notebooks' token with 'write' access, that you can then easily reuse for all\nnotebooks. </center>"
+          }
+        },
+        "15fdfbc426924d5091533d4462c9b53b": {
+          "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": "center",
+            "align_self": null,
+            "border": null,
+            "bottom": null,
+            "display": "flex",
+            "flex": null,
+            "flex_flow": "column",
+            "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": "50%"
+          }
+        },
+        "44aa945dc5e743b8a6aa3f3078caac77": {
+          "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
+          }
+        },
+        "fa8d2f6085ab4a31b61270aa6a1ea4c5": {
+          "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": ""
+          }
+        },
+        "7f3d379eb619439c98379c9b4f0795ec": {
+          "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
+          }
+        },
+        "6b048e6506aa454b9c5858828fbcd40c": {
+          "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": ""
+          }
+        },
+        "4690ab39fdb348359cb42526a3ebeb86": {
+          "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
+          }
+        },
+        "0c3bc5d3a65d4a8980e321caf5a5d4f6": {
+          "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": ""
+          }
+        },
+        "b5ea618a490448b4aa95637a03f9adb0": {
+          "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
+          }
+        },
+        "379537673a7b46a8bdcdbfdf346be8dd": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "ButtonStyleModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "ButtonStyleModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/base",
+            "_view_module_version": "1.2.0",
+            "_view_name": "StyleView",
+            "button_color": null,
+            "font_weight": ""
+          }
+        },
+        "bef797322e0940a0a4b0116bc502480d": {
+          "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
+          }
+        },
+        "8f9941f463cc4e7688daf9d70fb8937e": {
+          "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": ""
+          }
+        },
+        "9502b7944c4948cd9850aa515ba5f2d3": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "LabelModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "LabelModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "LabelView",
+            "description": "",
+            "description_tooltip": null,
+            "layout": "IPY_MODEL_56eeae6e18784dd2903addb05112e495",
+            "placeholder": "​",
+            "style": "IPY_MODEL_8bc3ad27022a45c29c0d7eac1da9956f",
+            "value": "Connecting..."
+          }
+        },
+        "56eeae6e18784dd2903addb05112e495": {
+          "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
+          }
+        },
+        "8bc3ad27022a45c29c0d7eac1da9956f": {
+          "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": ""
+          }
+        },
+        "9fec786322d047299a289cb80487ed52": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "LabelModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "LabelModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "LabelView",
+            "description": "",
+            "description_tooltip": null,
+            "layout": "IPY_MODEL_77dfd24b45f24da09dfe0c426d741aa7",
+            "placeholder": "​",
+            "style": "IPY_MODEL_c5313e81a1784396afd3cdc03f0920ea",
+            "value": "Token is valid (permission: write)."
+          }
+        },
+        "99801da567c5403284ac7555ea967dd3": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "LabelModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "LabelModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "LabelView",
+            "description": "",
+            "description_tooltip": null,
+            "layout": "IPY_MODEL_65e11d956f2b4f59a9ccc26176cd4a58",
+            "placeholder": "​",
+            "style": "IPY_MODEL_452967ad3cd94a21bcde37e3ea2784eb",
+            "value": "Your token has been saved in your configured git credential helpers (store)."
+          }
+        },
+        "d2dfcff538f54c449386cc843c2e7c6d": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "LabelModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "LabelModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "LabelView",
+            "description": "",
+            "description_tooltip": null,
+            "layout": "IPY_MODEL_9fbe272599ed450e840a918a8b6a1fbc",
+            "placeholder": "​",
+            "style": "IPY_MODEL_d3c9c4c34afd4ebfb8b911c2aac756b5",
+            "value": "Your token has been saved to /root/.cache/huggingface/token"
+          }
+        },
+        "7728b2f00d96487a9397a55c3cd336ce": {
+          "model_module": "@jupyter-widgets/controls",
+          "model_name": "LabelModel",
+          "model_module_version": "1.5.0",
+          "state": {
+            "_dom_classes": [],
+            "_model_module": "@jupyter-widgets/controls",
+            "_model_module_version": "1.5.0",
+            "_model_name": "LabelModel",
+            "_view_count": null,
+            "_view_module": "@jupyter-widgets/controls",
+            "_view_module_version": "1.5.0",
+            "_view_name": "LabelView",
+            "description": "",
+            "description_tooltip": null,
+            "layout": "IPY_MODEL_1852d779e7c8448d87ce381d4348868d",
+            "placeholder": "​",
+            "style": "IPY_MODEL_6a2cbfb5e7b64c8cac850d0b05ce8f47",
+            "value": "Login successful"
+          }
+        },
+        "77dfd24b45f24da09dfe0c426d741aa7": {
+          "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
+          }
+        },
+        "c5313e81a1784396afd3cdc03f0920ea": {
+          "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": ""
+          }
+        },
+        "65e11d956f2b4f59a9ccc26176cd4a58": {
+          "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
+          }
+        },
+        "452967ad3cd94a21bcde37e3ea2784eb": {
+          "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": ""
+          }
+        },
+        "9fbe272599ed450e840a918a8b6a1fbc": {
+          "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
+          }
+        },
+        "d3c9c4c34afd4ebfb8b911c2aac756b5": {
+          "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": ""
+          }
+        },
+        "1852d779e7c8448d87ce381d4348868d": {
+          "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
+          }
+        },
+        "6a2cbfb5e7b64c8cac850d0b05ce8f47": {
+          "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": ""
+          }
+        },
+        "d2ce25900eae4b41b8107401f174b78d": {
+          "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_84433208e40d4271b4db101602d267ef",
+              "IPY_MODEL_c58122cb38ef4486a96966ac861cf913",
+              "IPY_MODEL_121add4c3f014c04926701a1da4782f0"
+            ],
+            "layout": "IPY_MODEL_74c8e8e472904e6c9651509d8bfc6a50"
+          }
+        },
+        "84433208e40d4271b4db101602d267ef": {
+          "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_9ada60f4d6374e3ab1d616970fea8b57",
+            "placeholder": "​",
+            "style": "IPY_MODEL_7ce21dd248a047fe8b4a8be37a05bfac",
+            "value": "model.safetensors: 100%"
+          }
+        },
+        "c58122cb38ef4486a96966ac861cf913": {
+          "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_a3a9ece98352459baa56c051886a8924",
+            "max": 497774208,
+            "min": 0,
+            "orientation": "horizontal",
+            "style": "IPY_MODEL_ea47214f28704e6ea43cb064bf9f5294",
+            "value": 497774208
+          }
+        },
+        "121add4c3f014c04926701a1da4782f0": {
+          "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_c43afb5ad3ed429285c3a6d871a734d7",
+            "placeholder": "​",
+            "style": "IPY_MODEL_62a83b49f10f49ee9a8b85ad601b45c0",
+            "value": " 498M/498M [00:17&lt;00:00, 26.5MB/s]"
+          }
+        },
+        "74c8e8e472904e6c9651509d8bfc6a50": {
+          "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
+          }
+        },
+        "9ada60f4d6374e3ab1d616970fea8b57": {
+          "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
+          }
+        },
+        "7ce21dd248a047fe8b4a8be37a05bfac": {
+          "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": ""
+          }
+        },
+        "a3a9ece98352459baa56c051886a8924": {
+          "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
+          }
+        },
+        "ea47214f28704e6ea43cb064bf9f5294": {
+          "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": ""
+          }
+        },
+        "c43afb5ad3ed429285c3a6d871a734d7": {
+          "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
+          }
+        },
+        "62a83b49f10f49ee9a8b85ad601b45c0": {
+          "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