Update README.md

README.md

@@ -79,7 +79,7 @@ test_tokenized_dataset.set_format(type='torch', columns=['input_ids', 'attention
 # Load the model from the specified directory
 pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, device=0)
 # Save the predictions to the output file
-predictions_test = pipe(test_tokenized_dataset['text'], truncation=True, max_length=256)
+predictions_test = pipe(test_texts, truncation=True, max_length=256)
 actual_labels_test = []
 for prediction in predictions_test:
     pred_label = prediction['label']