Update README.md

README.md

@@ -40,12 +40,12 @@ 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 instance segmentation
 processor = AutoImageProcessor.from_pretrained("facebook/mask2former-swin-small-coco-instance")
-model = AutoForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-small-coco-instance")
+model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2former-swin-small-coco-instance")
 
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)

preprocessor_config.json

@@ -9,7 +9,7 @@
     0.4560000002384186,
     0.4059999883174896
   ],
-  "image_processor_type": "MaskFormerImageProcessor",
+  "image_processor_type": "Mask2FormerImageProcessor",
   "image_std": [
     0.2290000021457672,
     0.2239999920129776,