Enable syntax highlighting for example code

README.md

@@ -28,7 +28,8 @@ Let me know!
 Phi does not support device_map "auto", and does not seem to want to inference in fp16, so use bf16.
 
 Here is working code to inference, though it can be improved:
-```
+
+```python
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer