Fix code

README.md

@@ -95,6 +95,7 @@ python3 zipnn_decompress_path.py --path .
 
 Now just run the local version of the model.
 # Use a pipeline as a high-level helper
+```python
 from transformers import pipeline
 
 messages = [
@@ -102,12 +103,15 @@ messages = [
 ]
 pipe = pipeline("text-generation", model="PATH_TO_MODEL") # "." if in directory
 pipe(messages)
+```
 
 # Load model directly
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("PATH_TO_MODEL") # "." if in directory
 model = AutoModelForCausalLM.from_pretrained("PATH_TO_MODEL") # "." if in directory
+```
 
 ## Prompt Template