Update README.md

README.md

@@ -43,11 +43,11 @@ We will be rolling out intermediate checkpoints following the below schedule. We
 #### How to use
 You will need the transformers>=4.31
 Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
-```
+```python
 from transformers import AutoTokenizer
 import transformers 
 import torch
-model = "PY007/TinyLlama-1.1B-step-50K-105b"
+model = "TinyLlama/TinyLlama-1.1B-step-50K-105b"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",