Update README.md

README.md

@@ -4,7 +4,7 @@ tags:
 - vision
 - image-segmentation
 datasets:
-- scene_parse_150
+- coco
 widget:
 - src: http://images.cocodataset.org/val2017/000000039769.jpg
   example_title: Cats
@@ -14,7 +14,7 @@ widget:
 
 # MaskFormer
 
-MaskFormer model trained on COCO panoptic segmentation (base-sized version, Swin backbone). 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 panoptic segmentation (small-sized version, Swin backbone). 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 MaskFormer did not write a model card for this model so this model card has been written by the Hugging Face team.