lacuric commited on
Commit
78b0433
1 Parent(s): 8d2e4ef

Updated readme

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -5,7 +5,7 @@ language:
5
 
6
  # Text Classification GoEmotions
7
 
8
- This model 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 quantized version in ONNX format can be found [here](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw-onnx).
10
 
11
  The model with two labels only (toxicity and severe toxicity) is [here](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw-lite)
@@ -40,5 +40,5 @@ The following hyperparameters were used during training:
40
 
41
  # Deployment
42
 
43
- 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 and light resource utilization.
44
 
 
5
 
6
  # Text Classification GoEmotions
7
 
8
+ This model 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 quantized version in ONNX format can be found [here](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw-onnx).
10
 
11
  The model with two labels only (toxicity and severe toxicity) is [here](https://huggingface.co/minuva/MiniLMv2-toxic-jigsaw-lite)
 
40
 
41
  # Deployment
42
 
43
+ 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.
44