Update README.md (#118)

- Update README.md (02bf128f36f0a0e3aace3c8f25fec1e07b468db7)
- Update README.md (21014479dba59d2ee7bd508cee9e2463eaf18732)


Co-authored-by: Matthew Carrigan <Rocketknight1@users.noreply.huggingface.co>

README.md

@@ -316,26 +316,20 @@ messages = [
     {"role": "user", "content": "Who are you?"},
 ]
 
-prompt = pipeline.tokenizer.apply_chat_template(
-		messages, 
-		tokenize=False, 
-		add_generation_prompt=True
-)
-
 terminators = [
     pipeline.tokenizer.eos_token_id,
     pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
 ]
 
 outputs = pipeline(
-    prompt,
+    messages,
     max_new_tokens=256,
     eos_token_id=terminators,
     do_sample=True,
     temperature=0.6,
     top_p=0.9,
 )
-print(outputs[0]["generated_text"][len(prompt):])
+print(outputs[0]["generated_text"][-1])
 ```
 
 #### Transformers AutoModelForCausalLM