Update README.md

README.md

@@ -20,6 +20,7 @@ Label map:
 "B-attraction": 5
 "I-attraction": 6
 
+```python
 from transformers import AutoConfig, AutoModelForTokenClassification, BertTokenizer
 from transformers import TokenClassificationPipeline
 import json
@@ -39,8 +40,10 @@ tokenizer = BertTokenizer.from_pretrained(
   model_path,
 )
 
+# device=-1: cpu, device=0: gpu
 pipeline = TokenClassificationPipeline(model, tokenizer, device=-1)
 
 tokens = pipeline(["i want to book the hilton hotel near china town.", "can you reserve A & B restaurant for me?"])
+```
 
 Credit: Jia-Chen Jason Gu, Wilson Tam