Updated readme with code formatting

README.md

@@ -38,7 +38,7 @@ Further, the model is trained to generate questions involving complex scenarios
 An example notebook is provided [here](https://huggingface.co/NousResearch/Genstruct-7B/blob/main/notebook.ipynb), which details how to load and sample from the model.
 
 Alternatively, here's a minimal example:
-```
+```py
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 MODEL_NAME = 'NousResearch/Genstruct-7B'