Update README.md

README.md

@@ -128,16 +128,15 @@ We simply discard the system prompts.
 
 **To put it all together, the text before tokenization looks like this:**
 
-`general_instruction_response_text = "<|begin_of_text|>{question} {response}<|end_of_text|>"`
-
-or
-
-`instruction_augmented_text = "<|begin_of_text|>{instruction augmented text}<|end_of_text|>"`
+```python
+general_instruction_response_text = "<|begin_of_text|>{question} {response}<|end_of_text|>"
 
+instruction_augmented_text = "<|begin_of_text|>{instruction augmented text}<|end_of_text|>"
+```
 Then, for tokenization, you don't need to add BOS and EOS token ids. The tokenization code looks like this:
-
-`text_ids = tokenizer(text, add_special_tokens=False, **kwargs).input_ids`
-
+```python
+text_ids = tokenizer(text, add_special_tokens=False, **kwargs).input_ids
+```
 
 ## Citation
 If you find our work helpful, please cite us: