Update README.md

README.md

@@ -1,5 +1,9 @@
 ---
 license: mit
+base_model:
+- Ultralytics/YOLO11
+tags:
+- chemistry
 ---
 
 # Molecule Detection YOLO in MolParser
@@ -13,7 +17,7 @@ We provide several [ultralytics YOLO11]((https://github.com/ultralytics/ultralyt
 ![image/png](https://cdn-uploads.huggingface.co/production/uploads/65f7f16fb6941db5c2e7c4bf/7oWPoPxuEXSangDWnJ7mv.png)
 
 
-## General molecule structure detection models
+## 1⃣️ [MolDet-General] General molecule structure detection models
 
 `moldet_yolo11[size]_640_general.pt`
 
@@ -40,7 +44,7 @@ model = YOLO("moldet_yolo11l_640_general.pt")
 model.predict("path/to/image.png", save=True, imgsz=640, conf=0.5)
 ```
 
-## PDF molecule structure detection models
+## 2⃣️ [MolDet-Doc] PDF molecule structure detection models
 
 `moldet_yolo11[size]_960_doc.pt`
 
@@ -68,4 +72,15 @@ model = YOLO("moldet_yolo11l_960_doc.pt")
 model.predict("path/to/pdf_page_image.png", save=True, imgsz=960, conf=0.5)
 ```
 
+## 📖 Citation
 
+If you use this model in your work, please cite:
+
+```
+@article{fang2024molparser,
+  title={Molparser: End-to-end visual recognition of molecule structures in the wild},
+  author={Fang, Xi and Wang, Jiankun and Cai, Xiaochen and Chen, Shangqian and Yang, Shuwen and Tao, Haoyi and Wang, Nan and Yao, Lin and Zhang, Linfeng and Ke, Guolin},
+  journal={arXiv preprint arXiv:2411.11098},
+  year={2024}
+}
+```