Update README.md

README.md

@@ -85,7 +85,7 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 # Prepare Prompt
-prompt = "Once upon a time, Lily found a mysterious blue key."
+prompt = "Once upon a time, Tom found a blue car."
 inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
 
 # Generate Story
@@ -95,7 +95,12 @@ output = model.generate(
     temperature=0.7, 
     top_p=0.9,
     do_sample=True,
-    repetition_penalty=1.1
+    repetition_penalty=1.1,
+    # CRITICAL ADDITIONS BELOW:
+    eos_token_id=tokenizer.eos_token_id, 
+    pad_token_id=tokenizer.pad_token_id  
 )
 
+# Set skip_special_tokens=False first just to verify if the token is there(|endoftext|) 
+# then switch back to True for a clean output.
 print(tokenizer.decode(output[0], skip_special_tokens=True))