Commit
·
2fbbcea
1
Parent(s):
3a02dd4
Update README.md
Browse files
README.md
CHANGED
@@ -27,7 +27,8 @@ tokenizer = AutoTokenizer.from_pretrained('Deepchecks/roberta_base_formality_ran
|
|
27 |
model = ORTModelForSequenceClassification.from_pretrained('Deepchecks/roberta_base_formality_ranker_onnx')
|
28 |
|
29 |
# prepare the pipeline and generate inferences
|
30 |
-
|
|
|
31 |
res = pip(user_inputs, batch_size=64, truncation="only_first")
|
32 |
|
33 |
```
|
|
|
27 |
model = ORTModelForSequenceClassification.from_pretrained('Deepchecks/roberta_base_formality_ranker_onnx')
|
28 |
|
29 |
# prepare the pipeline and generate inferences
|
30 |
+
user_inputs = ["I hope this email finds you well", "I hope this email find you swell", "What's up doc?"]
|
31 |
+
pip = pipeline(task='text-classification', model=model, tokenizer=tokenizer, device=device, accelerator="ort")
|
32 |
res = pip(user_inputs, batch_size=64, truncation="only_first")
|
33 |
|
34 |
```
|