asofter commited on
Commit
9cbae18
·
verified ·
1 Parent(s): d44dffd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -35,6 +35,7 @@ from transformers import AutoTokenizer, pipeline
35
 
36
 
37
  tokenizer = AutoTokenizer.from_pretrained("laiyer/deberta-v3-large-zeroshot-v1-onnx")
 
38
  model = ORTModelForSequenceClassification.from_pretrained("laiyer/deberta-v3-large-zeroshot-v1-onnx")
39
  classifier = pipeline(
40
  task="zero-shot-classification",
 
35
 
36
 
37
  tokenizer = AutoTokenizer.from_pretrained("laiyer/deberta-v3-large-zeroshot-v1-onnx")
38
+ tokenizer.model_input_names = ["input_ids", "attention_mask"]
39
  model = ORTModelForSequenceClassification.from_pretrained("laiyer/deberta-v3-large-zeroshot-v1-onnx")
40
  classifier = pipeline(
41
  task="zero-shot-classification",