Update README.md
Browse files
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-
|
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)
|