From fc35ceb93c0f40b977506a2012da8f962a37052d Mon Sep 17 00:00:00 2001
From: Ludovic Moncla <moncla.ludovic@gmail.com>
Date: Mon, 16 Jan 2023 08:34:55 +0100
Subject: [PATCH] Update Classification_Zero-Shot-Learning.ipynb

---
 notebooks/Classification_Zero-Shot-Learning.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/notebooks/Classification_Zero-Shot-Learning.ipynb b/notebooks/Classification_Zero-Shot-Learning.ipynb
index f6050c8..4a7ab88 100644
--- a/notebooks/Classification_Zero-Shot-Learning.ipynb
+++ b/notebooks/Classification_Zero-Shot-Learning.ipynb
@@ -102,7 +102,7 @@
       "outputs": [],
       "source": [
         "import pandas as pd\n",
-        "\n",
+        "from tqdm import tqdm\n",
         "from transformers import BartForSequenceClassification, BartTokenizer\n"
       ]
     },
@@ -759,7 +759,7 @@
         "pred_labels = []\n",
         "prob_labels = []\n",
         "\n",
-        "for content in df[column_text].tolist():\n",
+        "for content in tqdm(df[column_text].tolist()):\n",
         "\n",
         "    true_probs = zero_shot_prediction(content[:1024], classes)\n",
         "    \n",
-- 
GitLab