Update README.md

README.md

@@ -7,7 +7,17 @@ tags:
 - svg
 ---
 
-This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+# DesigNet: Learning to Draw Vector Graphics as Designers Do
+
+[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/TomasGuija/DesigNet/blob/main/demo_designet_inference.ipynb) [![arXiv](https://img.shields.io/badge/arXiv-2604.06494-b31b1b.svg)](https://arxiv.org/abs/2604.06494)
+
+<p align="center">
+  <img src="https://github.com/TomasGuija/DesigNet/tree/main/docs/imgs/overview.jpg" width="500"/>
+</p>
+
+
 - Code: https://github.com/TomasGuija/DesigNet
 - Paper: https://arxiv.org/abs/2604.06494
-- Docs: [More Information Needed]
+- Docs: [More Information Needed]
+
+AI-driven content generation has made remarkable progress in recent years. However, neural networks and human designers operate in fundamentally different ways, making collaboration between them challenging. We address this gap for Scalable Vector Graphics (SVG) by equipping neural networks with tools commonly used by designers, such as axis alignment and explicit continuity control at command junctions. We introduce DesigNet, a hierarchical Transformer-VAE that operates directly on SVG sequences with a continuous command parameterization. Our main contributions are two differentiable modules: a continuity self-refinement module that predicts C^0, G^1, and C^1 continuity for each curve point and enforces it by modifying Bézier control points, and an alignment self-refinement module with snapping capabilities for horizontal or vertical lines. DesigNet produces editable outlines and achieves competitive results against state-of-the-art methods, with notably higher accuracy in continuity and alignment. These properties ensure the outputs are easier to refine and integrate into professional design workflows