Update README.md

README.md

@@ -21,7 +21,7 @@ Data retrieved from: https://zenodo.org/records/1162886
 
 - **Developers:** IBM Research
 - **GitHub Repository:** https://github.com/BiomedSciAI/biomed-multi-alignment
-- **Paper:** TBD
+- **Paper:** https://arxiv.org/abs/2410.22367
 - **Release Date**: Oct 28th, 2024
 - **License:** [Apache 2.0](https://www.apache.org/licenses/LICENSE-2.0).
 
@@ -45,6 +45,7 @@ from mammal.model import Mammal
 
 # Load Model
 model = Mammal.from_pretrained("ibm/biomed.omics.bl.sm.ma-ted-458m.protein_solubility")
+model.eval()
 
 # Load Tokenizer
 tokenizer_op = ModularTokenizerOp.from_pretrained("ibm/biomed.omics.bl.sm.ma-ted-458m.protein_solubility")
@@ -84,10 +85,13 @@ For more advanced usage, see our detailed example at: on `https://github.com/Bio
 
 If you found our work useful, please consider giving a star to the repo and cite our paper:
 ```
-@article{TBD,
-  title={TBD},
-  author={IBM Research Team},
-  jounal={arXiv preprint arXiv:TBD},
-  year={2024}
+@misc{shoshan2024mammalmolecularaligned,
+      title={MAMMAL -- Molecular Aligned Multi-Modal Architecture and Language}, 
+      author={Yoel Shoshan and Moshiko Raboh and Michal Ozery-Flato and Vadim Ratner and Alex Golts and Jeffrey K. Weber and Ella Barkan and Simona Rabinovici-Cohen and Sagi Polaczek and Ido Amos and Ben Shapira and Liam Hazan and Matan Ninio and Sivan Ravid and Michael M. Danziger and Joseph A. Morrone and Parthasarathy Suryanarayanan and Michal Rosen-Zvi and Efrat Hexter},
+      year={2024},
+      eprint={2410.22367},
+      archivePrefix={arXiv},
+      primaryClass={q-bio.QM},
+      url={https://arxiv.org/abs/2410.22367}, 
 }
 ```