chainyo commited on
Commit
074eaaf
1 Parent(s): 031ae9f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -102,7 +102,7 @@ result
102
  from optimum.onnxruntime import ORTModelForSequenceClassification
103
  from transformers import AutoTokenizer, pipeline
104
 
105
- HUB_MODEL = "cmarkea/distilcamembert-base-nli"
106
 
107
  tokenizer = AutoTokenizer.from_pretrained(HUB_MODEL)
108
  model = ORTModelForSequenceClassification.from_pretrained(HUB_MODEL)
102
  from optimum.onnxruntime import ORTModelForSequenceClassification
103
  from transformers import AutoTokenizer, pipeline
104
 
105
+ HUB_MODEL = "cmarkea/distilcamembert-base-sentiment"
106
 
107
  tokenizer = AutoTokenizer.from_pretrained(HUB_MODEL)
108
  model = ORTModelForSequenceClassification.from_pretrained(HUB_MODEL)