Update README.md (#5)

- Update README.md (d7338a97200143ba7522c960467c91ec3308d26c)


Co-authored-by: Mishig Davaadorj <[email protected]>

README.md

@@ -12,7 +12,10 @@ See details on the [AutoCoder GitHub](https://github.com/bin123apple/AutoCoder).
 
 Simple test script:
 
-```
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+from datasets import load_dataset
+
 model_path = ""
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path,