diff --git a/README.md b/README.md
index 0311256..68abec1 100644
--- a/README.md
+++ b/README.md
@@ -40,11 +40,11 @@ Here is how to use this model:
 import requests
 import torch
 from PIL import Image
-from transformers import Mask2FormerImageProcessor, Mask2FormerForUniversalSegmentation
+from transformers import AutoImageProcessor, Mask2FormerForUniversalSegmentation
 
 
 # load Mask2Former fine-tuned on COCO panoptic segmentation
-processor = Mask2FormerImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
+processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
 model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-base-coco-panoptic")
 
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"