Update model card

README.md

@@ -1,3 +1,14 @@
+---
+license: apache-2.0
+language:
+- en
+pipeline_tag: image-segmentation
+library_name: transformers
+tags:
+- wire
+- drone
+- computer_vision
+---
 # WireSegHR (Segmentation Only)
 
 This repository contains the segmentation-only implementation of the two-stage [WireSegHR model](https://arxiv.org/abs/2304.00221), training on the WireSegHR dataset plus the [TTPLA dataset](https://github.com/R3ab/ttpla_dataset).
@@ -150,4 +161,4 @@ Input:
 <img src="images/026_input.png" alt="Input" width="50%"/>
 
 Fine prediction:
-<img src="images/026_fine_pred.png" alt="Fine prediction" width="50%"/>
+<img src="images/026_fine_pred.png" alt="Fine prediction" width="50%"/>