Update README.md.

Author: msafari18

README.md

@@ -1,13 +1,53 @@
 # BarcodeMAE
 
-A pre-trained transformer model for inference on insect DNA barcoding data.
+A PyTorch implementation of BarcodeMAE, a model for enhancing DNA foundation models to address masking inefficiencies.
 
 <p align="center">
   <img src ="Figures/Arch_mae.png" alt="drawing" width="800"/>
 </p>
 
 * Check out our [paper](https://arxiv.org/pdf/2502.18405)
 
-#### Model checkpoint is available here: 
+#### Model checkpoint is available here: [BarcodeMAE](https://drive.google.com/file/d/18TqKC_gLYYDZEFfkMBRvWTHTT8Vb74Wv/view?usp=drive_link)
 
+### Using the model
 
+To be completed...
+
+### Reproducing the results from the paper
+
+0. Clone this repository and install the required libraries by running
+
+```shell
+pip install -e .
+```
+
+1. Download the [BIOSCAN-5M dataet from the git repo](https://github.com/bioscan-ml/BIOSCAN-5M)
+
+2. Pretrain BarcodeMAE
+
+To be completed...
+
+
+## Citation
+
+If you find BarcodeMAE useful in your research please consider citing:
+
+```bibtex
+@article{safari2025barcodemae,
+  title={Enhancing DNA Foundation Models to Address Masking Inefficiencies},
+  author={Monireh Safari
+    and Pablo Millan Arias
+    and Scott C. Lowe
+    and Lila Kari
+    and Angel X. Chang
+    and Graham W. Taylor
+  },
+  journal={arXiv preprint arXiv:2502.18405}
+  year={2025},
+  eprint={2502.18405},
+  archivePrefix={arXiv},
+  primaryClass={cs.LG},
+  doi={10.48550/arxiv.2311.02401},
+}
+```