Update README.md

README.md

@@ -6,6 +6,6 @@ model: PY007/TinyLlama-1.1B-intermediate-step-715k-1.5T
 ---
 TinyLLama 1.5T checkpoint trained to answer questions.
 ```
-f"{example['prompt'][i]}\n{example['completion'][i]}\n<END>"
+f"{example['prompt']}\n{example['completion']}\n<END>"
 ```
 No input/output, just question, then newline to begin the answer.