wilsontam commited on
Commit
bc0afb1
1 Parent(s): b8b0d37

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -21,7 +21,7 @@ def single_test(tokenizer, title_body_pair):
21
  return True if predictions == 0 else False
22
 
23
  if __name__ == '__main__':
24
- from transformers import AutoConfig, AutoTokenizer, AutoModel, AutoModelForSequenceClassification
25
 
26
  model_name = "wilsontam/bert-base-uncased-dstc10-kb-title-body-validate"
27
 
 
21
  return True if predictions == 0 else False
22
 
23
  if __name__ == '__main__':
24
+ from transformers import AutoConfig, AutoTokenizer, AutoModelForSequenceClassification
25
 
26
  model_name = "wilsontam/bert-base-uncased-dstc10-kb-title-body-validate"
27