Update README.md

README.md

@@ -79,4 +79,7 @@ predicted_class_ids = torch.argmax(logits, dim=-1)
 # Print the predicted class for each sequence
 for sequence, predicted_class in zip(new_sequences, predicted_class_ids):
     print(f"Sequence: {sequence}, Predicted class: {predicted_class.item()}")
-```
+```
+
+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).