Update README.md

README.md

@@ -17,6 +17,17 @@ metrics:
 ---
 # ESM-2 Fine-tuned CAFA-5
 
+## ESM-2 for Protein Function Prediction
+
+This is an experimental model fine-tuned from the 
+[esm2_t6_8M_UR50D](https://huggingface.co/facebook/esm2_t6_8M_UR50D) model 
+for multi-label classification. In particular, the model is fine-tuned on the CAFA-5 protein sequence dataset available 
+[here](https://huggingface.co/datasets/AmelieSchreiber/cafa_5). More precisely, the `train_sequences.fasta` file is the 
+list of protein sequences that were trained on, and the 
+`train_terms.tsv` file contains the gene ontology protein function labels for each protein sequence. For more details on using 
+ESM-2 models for multi-label sequence classification, [see here](https://huggingface.co/docs/transformers/model_doc/esm). 
+Due to the potentially complicated class weighting necessary for the hierarchical ontology, further fine-tuning will be necessary. 
+
 ## Training 
 
 Macro