Update README.md

README.md

@@ -1,10 +1,6 @@
 ---
-task_categories:
-- image-segmentation
-task_ids:
-- instance-segmentation
-pretty_name: XAMI
 tags:
+- instance-segmentation
 - Vision Transformers
 - CNN
 license: mit
@@ -12,12 +8,27 @@ license: mit
 
 # XAMI (XMM-Newton optical Artefact Mapping for astronomical Instance segmentation)
 
-Check the [XAMI](https://github.com/ESA-Datalabs/XAMI)** repo for training. 
+Check the **[XAMI](https://github.com/ESA-Datalabs/XAMI)** repo for training. 
 
 ## Model Links
 
-| Model Name      |Link                                                                                      |
-|-----------------|------------------------------------------------------------------------------------------|
-| YOLOv8          |[yolov8_segm](https://huggingface.co/iulia-elisa/XAMI/blob/main/yolo_weights/best.pt)     |
-| MobileSAM       |[sam_vit](https://huggingface.co/iulia-elisa/XAMI/blob/main/sam_weights/sam_0_best.pth)   |
-| XAMI            |[xami_model](https://huggingface.co/iulia-elisa/XAMI/blob/main/yolo_sam_final.pth)        |
+<p align="center">
+  <table border="1" cellspacing="0" cellpadding="5">
+    <tr>
+      <th>Model Name</th>
+      <th>Link</th>
+    </tr>
+    <tr>
+      <td>YOLOv8</td>
+      <td><a href="https://huggingface.co/iulia-elisa/XAMI/blob/main/yolo_weights/best.pt">yolov8_segm</a></td>
+    </tr>
+    <tr>
+      <td>MobileSAM</td>
+      <td><a href="https://huggingface.co/iulia-elisa/XAMI/blob/main/sam_weights/sam_0_best.pth">sam_vit</a></td>
+    </tr>
+    <tr>
+      <td>XAMI</td>
+      <td><a href="https://huggingface.co/iulia-elisa/XAMI/blob/main/yolo_sam_final.pth">xami_model</a></td>
+    </tr>
+  </table>
+</p>