Update README.md

README.md

@@ -49,7 +49,8 @@ prompt = f"### Human: Please summarize the following article.\n\n{article}.\n###
 
 inputs = tokenizer(header + prompt, return_tensors="pt")
 sample = model.generate(**inputs, do_sample=True, max_new_tokens=2048, top_k=100, eos_token_id=50256)
-print(tokenizer.decode(sample[0]))
+output = tokenizer.decode(sample[0])
+print(output.strip().replace("Assistant:", ""))
 ```
 
 ## Citation