From 69472d84a60d61beed6deb6555df0fcf611c75c3 Mon Sep 17 00:00:00 2001
From: Alara Dirik <adirik@users.noreply.huggingface.co>
Date: Wed, 4 Jan 2023 14:44:52 +0000
Subject: [PATCH] Update README.md

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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"