pvl commited on
Commit
4da502e
·
verified ·
1 Parent(s): 888edab

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -5,7 +5,7 @@ language:
5
 
6
  # Text Classification Toxicity
7
 
8
- This is a quantized onnx model and is a fined-tuned version of [nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large) on the on the [Jigsaw 1st Kaggle competition](https://www.kaggle.com/competitions/jigsaw-toxic-comment-classification-challenge) dataset using [unitary/toxic-bert](https://huggingface.co/unitary/toxic-bert) as teacher model.
9
  The original model can be found [here](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw)
10
 
11
 
@@ -143,5 +143,4 @@ The following hyperparameters were used during training:
143
 
144
  # Deployment
145
 
146
- Check [this repository](https://github.com/minuva/toxicity-prediction-serverless) to see how to easily deploy this model in a serverless environment with fast CPU inference.
147
-
 
5
 
6
  # Text Classification Toxicity
7
 
8
+ This is a quantized onnx model and is a fined-tuned version of [MiniLMv2-L6-H384](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large) on the on the [Jigsaw 1st Kaggle competition](https://www.kaggle.com/competitions/jigsaw-toxic-comment-classification-challenge) dataset using [unitary/toxic-bert](https://huggingface.co/unitary/toxic-bert) as teacher model.
9
  The original model can be found [here](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw)
10
 
11
 
 
143
 
144
  # Deployment
145
 
146
+ Check out [fast-nlp-text-toxicity repository](https://github.com/minuva/fast-nlp-text-toxicity) for a FastAPI based server to deploy this model in CPU devices.