diff --git a/README.md b/README.md
index c706bff..7da6006 100644
--- a/README.md
+++ b/README.md
@@ -9,7 +9,7 @@ datasets:
 
 # MaskFormer
 
-MaskFormer model trained on ADE20k semantic segmentation. 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 ADE20k semantic segmentation (large-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.