Update app.py

app.py

@@ -22,7 +22,6 @@ kp = KeyPhraseTransformer()
 
 dataset = load_dataset("Unknown92/Resume_dataset")
 
-
 tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
 
 def tokenize_function(examples):
@@ -30,8 +29,8 @@ def tokenize_function(examples):
 
 tokenized_datasets = dataset.map(tokenize_function, batched=True)
 
-small_train_dataset = tokenized_datasets["train"].shuffle(seed=42).select(range(200))
-small_eval_dataset = tokenized_datasets["test"].shuffle(seed=42).select(range(200))
+small_train_dataset = tokenized_datasets["Train"].shuffle(seed=42).select(range(200))
+small_eval_dataset = tokenized_datasets["Test"].shuffle(seed=42).select(range(200))
 
 model = AutoModelForSequenceClassification.from_pretrained("bert-base-cased", num_labels=5)
 training_args = TrainingArguments(output_dir="test_trainer", evaluation_strategy="epoch")