Update README.md

README.md

@@ -9,10 +9,12 @@ pip install -U transformers
 ```
 
 ```python
+import random
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("docto/Docto-Bot")
 model = AutoModelForCausalLM.from_pretrained("docto/Docto-Bot")
+special_token = '<|endoftext|>'
 
 prompt_text = 'Question: I am having fever\nAnswer:'
 encoded_prompt = tokenizer.encode(prompt_text,
@@ -30,5 +32,5 @@ output_sequences = model.generate(
 )
 result = tokenizer.decode(random.choice(output_sequences))
 result = result[result.index("Answer: "):result.index(special_token)]
-print(result[2:])
+print(result[8:])
 ```