harsh-deepchecks commited on
Commit
2fbbcea
·
1 Parent(s): 3a02dd4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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
- pip = pipeline(task='text-classification', model=onnx_model, tokenizer=tokenizer, device=device, accelerator="ort")
 
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
  ```