Update README.md
Browse files
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,
|
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 |
|