Update README.md

README.md

@@ -25,11 +25,11 @@ The dataset used is [OpenAssistant/oasst_top1_2023-08-25](https://huggingface.co
 #### 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-Chat-v0.3"
+model = "TinyLlama/TinyLlama-1.1B-Chat-v0.3"
 tokenizer = AutoTokenizer.from_pretrained(model)
 pipeline = transformers.pipeline(
     "text-generation",