Update README.md

README.md

@@ -19,6 +19,7 @@ This is the model card of a Phi-2 model trained on a synthetic data set to solve
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
 ### Direct Use
+```python
 import torch
 from transformers import AutoModelForCausalLM, BitsAndBytesConfig, set_seed
 
@@ -26,7 +27,7 @@ from transformers import AutoModelForCausalLM, BitsAndBytesConfig, set_seed
 set_seed(42)
 
 # Load model
-```python
+
 modelpath = "DisgustingOzil/phi-2-riddler"
 model = AutoModelForCausalLM.from_pretrained(
     modelpath,