Update README.md

README.md

@@ -6,6 +6,7 @@ base_model: nvidia/mit-b0
 model-index:
 - name: Aerial-Drone-Image-Segmentation
   results: []
+pipeline_tag: image-segmentation
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -13,7 +14,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Aerial-Drone-Image-Segmentation
 
-This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
+This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0)
 It achieves the following results on the evaluation set:
 - Loss: 0.8852
 - Mean Iou: 0.2994
@@ -24,15 +25,6 @@ It achieves the following results on the evaluation set:
 
 More information needed
 
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
 
 ### Training hyperparameters
 
@@ -91,9 +83,12 @@ The following hyperparameters were used during training:
 | 0.8089        | 50.0  | 800  | 0.8852          | 0.2994   | 0.3923        | 0.7774           |
 
 
+
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/64fec5de57ccb8f1bdfbec54/UXq4-iUOSg2B5ff9tMezV.png)
+
 ### Framework versions
 
 - Transformers 4.38.1
 - Pytorch 2.1.2
 - Datasets 2.1.0
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2