Update README.md

README.md

@@ -20,4 +20,27 @@ Label map:
 "B-attraction": 5
 "I-attraction": 6
 
+from transformers import AutoConfig, AutoModelForTokenClassification, BertTokenizer
+from transformers import TokenClassificationPipeline
+import json
+
+model_path = "wilsontam/dstc9_ner"
+
+config = AutoConfig.from_pretrained(
+  model_path,
+  num_labels=7,
+)   
+model = AutoModelForTokenClassification.from_pretrained(
+  model_path,
+  from_tf=False,
+  config=config,
+)   
+tokenizer = BertTokenizer.from_pretrained(
+  model_path,
+)
+
+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