mjwong
/

contriever-msmarco-mnli

Zero-Shot Classification

text-classification

Inference Endpoints

Model card Files Files and versions Community

mjwong commited on Jul 22, 2023

Commit

3ae14ae

•

1 Parent(s): b8f30d7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -57,7 +57,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 # device = "cuda:0" or "cpu"
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-model_name = 'mjwong/contriever-msmarco-mnli'
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)

 # device = "cuda:0" or "cpu"
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+model_name = "mjwong/contriever-msmarco-mnli"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)