Wrap badges if space is not enough

app.py

@@ -14,17 +14,24 @@ MARKDOWN_DESCRIPTION = """
     <h1>Vocabulary-free Image Classification</h1>
 </div>
 
-<div style="display: flex; align-items: center; justify-content: center; margin-bottom: 1rem;">
-    <a href="https://github.com/altndrr/vic" style="margin-right: 0.5rem;">
+<div style="display: flex;
+            flex-wrap: wrap;
+            align-items: center;
+            justify-content: center;
+            margin-bottom: 1rem;">
+    <a href="https://github.com/altndrr/vic" style="margin-right: 0.5rem; margin-bottom: 0.5rem;">
         <img src="https://img.shields.io/badge/code-github.altndrr%2Fvic-blue.svg"/>
     </a>
-    <a href="https://huggingface.co/spaces/altndrr/vic" style="margin-right: 0.5rem;">
+    <a href="https://huggingface.co/spaces/altndrr/vic" style="margin-right: 0.5rem;
+                                                               margin-bottom: 0.5rem;">
         <img src="https://img.shields.io/badge/demo-hf.altndrr%2Fvic-yellow.svg"/>
     </a>
-    <a href="https://arxiv.org/abs/2306.00917" style="margin-right: 0.5rem;">
+    <a href="https://arxiv.org/abs/2306.00917" style="margin-right: 0.5rem;
+                                                      margin-bottom: 0.5rem;">
         <img src="https://img.shields.io/badge/paper-arXiv.2306.00917-B31B1B.svg"/>
     </a>
-    <a href="https://alessandroconti.me/papers/2306.00917" style="margin-right: 0.5rem;">
+    <a href="https://alessandroconti.me/papers/2306.00917" style="margin-right: 0.5rem;
+                                                                  margin-bottom: 0.5rem;">
         <img src="https://img.shields.io/badge/website-gh--pages.altndrr%2Fvic-success.svg"/>
     </a>
 </div>