Update README.md

README.md

@@ -13,8 +13,9 @@ arxiv: 2507.13077
 ---
 
 
-# ESMCBA — Download and Run With `embeddings.py`
+# ESMCBA 
 
+![Graph](https://github.com/sermare/ESMCBA/blob/main/figures_manuscript/graph_abstract2.png)
 
 **Code**: https://github.com/sermare/ESMCBA  
 **Models**: https://huggingface.co/smares/ESMCBA
@@ -62,7 +63,7 @@ git clone https://github.com/sermare/ESMCBA
 
 Inside the repo you should have `embeddings.py` available. If your file is named `embeddings_generation.py`, use that name instead in the commands below.
 
-## 3. Pick a model checkpoint
+## 3. Pick a model checkpoint (all models at the end of this Markdown)
 
 All checkpoints live in the model repo: `smares/ESMCBA`.
 
@@ -74,6 +75,9 @@ Examples of filenames:
 
 You can browse all files here: https://huggingface.co/smares/ESMCBA
 
+![SOTA](https://github.com/sermare/ESMCBA/blob/main/figures_manuscript/4a_SOTA_models-4.png)
+
+
 ## 4. Download a checkpoint
 
 ### Option A — download to a folder next to the code