Update README.md

README.md

@@ -15,15 +15,21 @@ For further information, please refer to https://github.com/yctam/dstc10_track2_
 ```python
 def single_test(tokenizer, title_body_pair):
     result = tokenizer([title_body_pair], return_tensors="pt")
-    result.to(device)
     model.eval()
     outputs = model(**result)
     predictions = outputs.logits.argmax(dim=-1)
     return True if predictions == 0 else False
 
 if __name__ == '__main__':
+    from transformers import AutoConfig, AutoTokenizer, AutoModel, AutoModelForSequenceClassification
+
+    model_name = "wilsontam/bert-base-uncased-dstc10-kb-title-body-validate"
+
+    config = AutoConfig.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
+    model = AutoModelForSequenceClassification.from_pretrained(".")
+
     sentence = "Can I check in anytime?"
     body = "Yes, 24 Hours Front Desk Avaliable."
-    pair = (sentence, body)
-    print(single_test(pair)) # Expect: True       
+    print(single_test((sentence, body))) # Expect: True       
 ```