Update README.md

README.md

@@ -17,9 +17,22 @@ language:
 
 ## Model Summary
 **CADFS-2B** is a fine-tuned version of [Qwen2-VL-2B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct) for parametric CAD generation. Given a **text description** or **multi-view images** of a 3D object, the model outputs a [FeatureScript](https://cad.onshape.com/FsDoc/) code that reconstructs the object as a fully parametric CAD model importable into [Onshape](https://www.onshape.com).
-CADFS-2B model weights. SFT of the Qwen2-VL-2B.
 
 ---
 
 ## Usage
-For inference code see the [CADFS GitHub repository](https://github.com/VladPyatov/CADFS).
+For inference code and FeatureScript to STEP file conversion please visit the [CADFS GitHub repository](https://github.com/VladPyatov/CADFS).
+
+---
+
+## Citation
+```bibtex
+@InProceedings{pyatov2026cadfs,
+    author    = {Pyatov, Vladislav and Bobrovskikh, Gleb and Galochkin, Saveliy and Boldyrev, Nikita and Voynov, Oleg and Filippov, Alexander and Ferrer, Gonzalo and Wonka, Peter and Burnaev, Evgeny},
+    title     = {CADFS: A Big CAD Program Dataset and Framework for Computer-Aided Design with Large Language Models},
+    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
+    month     = {June},
+    year      = {2026},
+    pages     = {10176-10186}
+}
+```