Update README.md

README.md

@@ -35,6 +35,6 @@ samples = model.generate(
     do_sample=False,
 )
 samples = tokenizer.batch_decode(samples, skip_special_tokens=True)
-for i, sample in enumerate(samples):
-    print(f"Prompt: {prompts[i]}\nSample: {sample}\n")
+for prompt, sample in zip(prompts, samples):
+    print(f"Prompt: {prompt}\nSample: {sample}\n")
 ```