Update README.md

README.md

@@ -111,7 +111,7 @@ If using dria_agent, system prompts and tooling work out of the box—no extra s
 
 You can use Tiny-Agent-a-3B directly with transformers:
 
-````
+````python
 import json
 from typing import Any, Dict, List
 from transformers import AutoModelForCausalLM, AutoTokenizer