Update README.md

README.md

@@ -21,10 +21,10 @@ model = MemoryLLM.from_pretrained("YuWangX/memoryllm-8b-chat")
 tokenizer = AutoTokenizer.from_pretrained("YuWangX/memoryllm-8b-chat")
 ```
 
-```python
-
 ### How to use the model
+
 Inject a piece of context into the model using the following script:
+
 ```python
 model = model.cuda()