Update README.md

README.md

@@ -15,6 +15,10 @@ 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`, `transport_proteins`, and `structural_proteins`. 
+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).
+
+## Using the Model
 To use the model, try running:
 
 ``` python3
@@ -81,6 +85,3 @@ predicted_class_ids = torch.argmax(logits, dim=-1)
 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).