Update README.md

README.md

@@ -86,11 +86,15 @@ logging.set_verbosity(logging.CRITICAL)
 
 pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
-prompt_template = "<|system|>\n<|end|>\n<|user|>\n{query}<|end|>\n<|assistant|>"
-prompt = prompt_template.format(query="How do I sort a list in Python?")
-# We use a special <|end|> token with ID 49155 to denote ends of a turn
-outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.2, top_k=50, top_p=0.95, eos_token_id=49155)
-# You can sort a list in Python by using the sort() method. Here's an example:\n\n```\nnumbers = [3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5]\nnumbers.sort()\nprint(numbers)\n```\n\nThis will sort the list in place and print the sorted list.
+prompt_template = '''Below is an instruction that describes a task. Write a response that appropriately completes the request
+
+### Instruction: {prompt}
+
+### Response:'''
+prompt = prompt_template.format(prompt="How do I sort a list in Python?")
+
+outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.2, top_k=50, top_p=0.95)
+
 print(outputs[0]['generated_text'])
 ```