Update README.md

README.md

@@ -25,9 +25,9 @@ This repository contains a version of [TeenyTinyLlama-162m]() fine-tuned on a tr
 
 import evaluate
 import numpy as np
-from datasets import load_dataset, Dataset, DatasetDict
+from datasets import load_dataset
 from transformers import AutoTokenizer, DataCollatorWithPadding
-from transformers import AutoModelForSequenceClassification, TrainingArguments, Traine
+from transformers import AutoModelForSequenceClassification, TrainingArguments, Trainer
 
 # Load the task
 dataset = load_dataset("christykoh/imdb_pt")