From 51c0c679410d5171f0870a6f459b0f9804bcade1 Mon Sep 17 00:00:00 2001
From: Robert Brisita <986796+rbrisita@users.noreply.github.com>
Date: Tue, 20 Aug 2024 03:57:18 -0400
Subject: [PATCH] Update README.md (#563)

Wrong dataset used (`small_train_dataset`) to create `small_eval_dataset`.
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 597bc8f5..5a4c50fd 100644
--- a/README.md
+++ b/README.md
@@ -282,7 +282,7 @@ small_train_dataset = dataset["train"].shuffle(seed=42).select(range(1000))
 small_eval_dataset = dataset["test"].shuffle(seed=42).select(range(300))
 
 small_train_dataset = small_train_dataset.map(tokenize_function, batched=True)
-small_eval_dataset = small_train_dataset.map(tokenize_function, batched=True)
+small_eval_dataset = small_eval_dataset.map(tokenize_function, batched=True)
 
 # download the model
 model = AutoModelForSequenceClassification.from_pretrained(