harsh-deepchecks commited on
Commit
fe2b41d
1 Parent(s): dc3621b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -0
README.md CHANGED
@@ -9,6 +9,13 @@ license: cc-by-nc-sa-4.0
9
  This model represents an ONNX-optimized version of the original [roberta_toxicity_classifier](https://huggingface.co/s-nlp/roberta_toxicity_classifier) model.
10
  It has been specifically tailored for GPUs and may exhibit variations in performance when run on CPUs.
11
 
 
 
 
 
 
 
 
12
  ## How to use
13
  ```python
14
  from transformers import AutoTokenizer
 
9
  This model represents an ONNX-optimized version of the original [roberta_toxicity_classifier](https://huggingface.co/s-nlp/roberta_toxicity_classifier) model.
10
  It has been specifically tailored for GPUs and may exhibit variations in performance when run on CPUs.
11
 
12
+ ## Dependencies
13
+
14
+ Please install the following dependency before you begin working with the model:
15
+ ```sh
16
+ pip install optimum[onnxruntime-gpu]
17
+ ```
18
+
19
  ## How to use
20
  ```python
21
  from transformers import AutoTokenizer