Update README.md

README.md

@@ -13,7 +13,7 @@ pip install git+https://github.com/etowahadams/interprot.git
 
 ## Usage
 
-### Load SAE
+Load the SAE
 ```python
 from safetensors.torch import load_file
 from interprot.sae_model import SparseAutoencoder
@@ -23,7 +23,7 @@ checkpoint_path = 'esm2_plm1280_l24_sae4096.safetensors'
 sae_model.load_state_dict(load_file(checkpoint_path))
 ```
 
-### ESM -> SAE Inference
+Load ESM and run ESM inference -> SAE inference
 ```
 import torch
 from transformers import AutoTokenizer, EsmModel