Update README.md

README.md

@@ -1,3 +1,29 @@
----
-license: cc-by-nc-4.0
----
+---
+license: cc-by-nc-4.0
+base_model:
+- Qwen/Qwen2-1.5B
+---
+
+### CAD-Recode: Reverse Engineering CAD Code from Point Clouds
+
+CAD-Recode is trained to transform point cloud to 3D CAD model as a valid Python code.
+CAD-Recode basically is Qwen2-1.5B + a single linear layer, which means that we share the original tokenizer of Qwen2.
+For minimal reproducible example please visit our github https://github.com/filaPro/cad-recode.
+And if you like it, give us a github 🌟.
+
+
+### Citation
+
+If you find this work useful for your research, please cite our paper:
+
+```
+@misc{rukhovich2024cadrecode,
+      title={CAD-Recode: Reverse Engineering CAD Code from Point Clouds}, 
+      author={Danila Rukhovich, Elona Dupont, Dimitrios Mallis, Kseniya Cherenkova, Anis Kacem, Djamila Aouada},
+      year={2024},
+      eprint={2412.14042},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2412.14042}, 
+}
+```