Update README.md

README.md

@@ -7,11 +7,11 @@ datasets:
 - coco
 ---
 
-# Mask
+# MaskFormer (small-sized, trained on COCO)
 
-Mask model trained on coco. It was introduced in the paper [Per-Pixel Classification is Not All You Need for Semantic Segmentation](https://arxiv.org/abs/2107.06278) and first released in [this repository](https://github.com/facebookresearch/MaskFormer/blob/da3e60d85fdeedcb31476b5edd7d328826ce56cc/mask_former/modeling/criterion.py#L169). 
+MaskFormer model trained on COCO. It was introduced in the paper [Per-Pixel Classification is Not All You Need for Semantic Segmentation](https://arxiv.org/abs/2107.06278) and first released in [this repository](https://github.com/facebookresearch/MaskFormer/blob/da3e60d85fdeedcb31476b5edd7d328826ce56cc/mask_former/modeling/criterion.py#L169). 
 
-Disclaimer: The team releasing Mask did not write a model card for this model so this model card has been written by the Hugging Face team.
+Disclaimer: The team releasing MaskFormer did not write a model card for this model so this model card has been written by the Hugging Face team.
 
 ## Model description
 
@@ -51,6 +51,4 @@ Here is how to use this model:
 >>> output = feature_extractor.post_process_panoptic_segmentation(outputs)
 ```
 
-
-
 For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/maskformer).