Add library_name and improve model card

README.md

@@ -1,10 +1,11 @@
 ---
-license: apache-2.0
-datasets:
-- Moreza009/AAV_datasets
 base_model:
 - facebook/esm2_t6_8M_UR50D
+datasets:
+- Moreza009/AAV_datasets
+license: apache-2.0
 pipeline_tag: text-classification
+library_name: transformers
 tags:
 - biology
 - protein
@@ -12,15 +13,17 @@ tags:
 - capsid_engineering
 ---
 
-This is an auxiliary model for the paper ********[AAVGen: Precision Engineering of Adeno-associated Virus for Renal Selective Targeting]()********
+This is an auxiliary model for the paper **[AAVGen: Precision Engineering of Adeno-associated Viral Capsids for Renal Selective Targeting](https://huggingface.co/papers/2602.18915)**.
+
+AAVGen is a generative artificial intelligence framework for de novo design of AAV capsids with enhanced multi-trait profiles. This specific model is one of the ESM-2-based regression predictors (based on [facebook/esm2_t6_8M_UR50D](https://huggingface.co/facebook/esm2_t6_8M_UR50D)) used to guide the generative process by predicting key properties such as production fitness, kidney tropism, and thermostability.
 
 Please visit: 
 <p align="center">
   <a href="https://github.com/mohammad-gh009/AAVGen">
-    <img src="https://img.shields.io/badge/GitHub-Code-blue.svg?logo=githu" alt="Github">
+    <img src="https://img.shields.io/badge/GitHub-Code-blue.svg?logo=github" alt="Github">
   </a>
   <a href="https://arxiv.org/abs/2602.18915">
-    <img src="https://img.shields.io/badge/arXiv-2602.18915-b31b1b.svg" alt="arXive">
+    <img src="https://img.shields.io/badge/arXiv-2602.18915-b31b1b.svg" alt="arXiv">
   </a>
   <a href="https://huggingface.co/datasets/Moreza009/AAV_datasets">
     <img src="https://img.shields.io/badge/🤗%20Hugging%20Face-Datasets-yellow" alt="HuggingFace datasets">
@@ -33,3 +36,19 @@ Please visit:
 <p align="center">
   <img src="https://raw.githubusercontent.com/mohammad-gh009/AAVGen/main/assets/Logo.png" alt="Logo" width="500">
 </p>
+
+## Citation
+
+If you use AAVGen in your research, please cite:
+
+```bibtex
+@misc{ghaffarzadehesfahani2026aavgenprecisionengineeringadenoassociated,
+      title={AAVGen: Precision Engineering of Adeno-associated Viral Capsids for Renal Selective Targeting}, 
+      author={Mohammadreza Ghaffarzadeh-Esfahani and Yousof Gheisari},
+      year={2026},
+      eprint={2602.18915},
+      archivePrefix={arXiv},
+      primaryClass={q-bio.QM},
+      url={https://arxiv.org/abs/2602.18915}, 
+}
+```