Update README.md

README.md

@@ -14,7 +14,7 @@ pipeline_tag: zero-shot-classification
 
 # ESM-2 Sequence Classifier
 This is a small sequence classifier trained on synthetic data generate by GPT-4 
-which classifies protein sequences into three categories `enzymes`, `receptor_proteins`, and `structural_proteins`. 
+which classifies protein sequences into three categories `enzymes` (class `0`), `receptor_proteins` (class `1`), and `structural_proteins` (class `2`). 
 This is trained using [facebook/esm2_t6_8M_UR50D](https://huggingface.co/facebook/esm2_t6_8M_UR50D), one of the 
 Meta AI [ESM-2 models](https://huggingface.co/docs/transformers/model_doc/esm).