Update README.md

README.md

@@ -13,6 +13,8 @@ For further information, please refer to https://github.com/yctam/dstc10_track2_
 
 ---
 ```python
+from transformers import AutoConfig, AutoTokenizer, AutoModelForSequenceClassification
+
 def single_test(tokenizer, title_body_pair):
     result = tokenizer([title_body_pair], return_tensors="pt")
     model.eval()
@@ -21,8 +23,6 @@ def single_test(tokenizer, title_body_pair):
     return True if predictions == 0 else False
 
 if __name__ == '__main__':
-    from transformers import AutoConfig, AutoTokenizer, AutoModelForSequenceClassification
-
     model_name = "wilsontam/bert-base-uncased-dstc10-kb-title-body-validate"
 
     config = AutoConfig.from_pretrained(model_name)