jmvcoelho
/

ad-classifier-v0.2

Text Classification

Model card Files Files and versions Community

jmvcoelho commited on Mar 2

Commit

29e7186

·

verified ·

1 Parent(s): 7dd33ba

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -13,7 +13,7 @@ Binary classification model for ad-detection on QA Systems.
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
-classifier_model_path = "jmvcoelho/ad-classifier-v0.1"
 tokenizer = AutoTokenizer.from_pretrained(classifier_model_path)
 model = AutoModelForSequenceClassification.from_pretrained(classifier_model_path)
 model.eval()

 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+classifier_model_path = "jmvcoelho/ad-classifier-v0.2"
 tokenizer = AutoTokenizer.from_pretrained(classifier_model_path)
 model = AutoModelForSequenceClassification.from_pretrained(classifier_model_path)
 model.eval()