Update README.md

README.md

@@ -61,6 +61,7 @@ Don't make up value not in the Input. Output must be a well-formed JSON object.`
 ```
 
 # Load model directly
+```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("mychen76/mistral7b_ocr_to_json_v1")
@@ -80,4 +81,4 @@ with torch.inference_mode():
     outputs = model.generate(**inputs, max_new_tokens=512) 
     result_text = tokenizer.batch_decode(outputs)[0]
     print(result_text)
-
+```