MoritzLaurer
/

ernie-m-base-mnli-xnli

Zero-Shot Classification

text-classification

Inference Endpoints

Model card Files Files and versions Community

MoritzLaurer HF staff commited on Mar 3, 2023

Commit

86cca6a

·

1 Parent(s): b975e32

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -66,7 +66,7 @@ device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cp
 model_name = "MoritzLaurer/ernie-m-base-mnli-xnli"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
 premise = "Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU"
 hypothesis = "Emmanuel Macron is the President of France"

 model_name = "MoritzLaurer/ernie-m-base-mnli-xnli"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name).to(device)
 premise = "Angela Merkel ist eine Politikerin in Deutschland und Vorsitzende der CDU"
 hypothesis = "Emmanuel Macron is the President of France"