CogComp
/

ZeroShotWiki

Text Classification

Model card Files Files and versions

hantian2 commited on May 14, 2022

Commit

0b85bb8

·

1 Parent(s): bef0342

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -15,7 +15,7 @@ Concatenate the text sentence with each of the candidate labels as input to the
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
 model = AutoModelForSequenceClassification.from_pretrained("CogComp/ZeroShotWiki")
 labels = ["sports", "business", "politics"]

 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
+tokenizer = AutoTokenizer.from_pretrained("CogComp/ZeroShotWiki")
 model = AutoModelForSequenceClassification.from_pretrained("CogComp/ZeroShotWiki")
 labels = ["sports", "business", "politics"]