Update README.md

README.md

@@ -14,9 +14,9 @@ tags:
 ---
 # ToxicityModelPT (Portuguese)
 
-The `ToxicityModelPT` is a modified BERT model that can be used to score the toxicity of completion to a given prompt. It is based on a [BERT model](https://huggingface.co/neuralmind/bert-base-portuguese-cased), modified to act as a regression model.
+The `ToxicityModelPT` is a modified BERT model that can be used to score the toxicity of a sentence (prompt + completion). It is based on a [BERT model](https://huggingface.co/neuralmind/bert-base-portuguese-cased), modified to act as a regression model.
 
-The `ToxicityModel` allows the specification of an `alpha` parameter, which is a multiplier to the toxicity score. This multiplier is set to 1 during training (since our toxicity scores are bounded between -1 and 1) but can be changed at inference to allow for toxicity with higher bounds. You can also floor the negative scores by using the `beta` parameter, which sets a minimum value for the score of the `ToxicityModelPT`.
+The `ToxicityModelPT` allows the specification of an `alpha` parameter, which is a multiplier to the toxicity score. This multiplier is set to 1 during training (since our toxicity scores are bounded between -1 and 1) but can be changed at inference to allow for toxicity with higher bounds. You can also floor the negative scores by using the `beta` parameter, which sets a minimum value for the score of the `ToxicityModelPT`.
 
 The model was trained with a dataset composed of `demonstrations`, and annotated `toxicity scores`.