Update README.md

README.md

@@ -45,6 +45,7 @@ tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, max_len=512)
 pipe = pipeline("text-classification", model=model, tokenizer=tokenizer, device=0)
  
 # See https://huggingface.co/docs/transformers/main_classes/pipelines#transformers.pipeline
-for out in tqdm(pipe(KeyDataset(dataset, "text"), padding=True, truncation=True)):
-   print(out)
+for i, out in enumerate(tqdm(pipe(KeyDataset(dataset, "text"), padding=True, truncation=True))):
+  print(dataset["text"][i])
+  print(out)
 ```