Update README.md

README.md

@@ -30,7 +30,9 @@ It achieves the following results on the validation set:
 * Precision: 0.7671
 
 # Description
-This model is based on an existing lighter variation of BERT (distilBERT), in order to predict different mental disorders. It does this based on texts or posts (from Reddit) about general experiences of users with mental health problems.
+This model is based on an existing lighter variation of BERT (distilBERT), in order to predict different mental disorders. It is trained on a costume dataset of texts or posts (from Reddit) about general experiences of users with mental health problems.
+All direct mentions of the disorder names in the texts were removed. 
+ 
 It includes the following classes: 
 
 * Borderline