Update predict.py

predict.py

@@ -10,13 +10,13 @@ os.system('pip install git+https://github.com/facebookresearch/detectron2.git')
 from detectron2.data import MetadataCatalog
 from detectron2.utils.visualizer import ColorMode, Visualizer
 from color_palette import ade_palette
-from transformers import MaskFormerImageProcessor, Mask2FormerForUniversalSegmentation
+from transformers import Mask2FormerImageProcessor, Mask2FormerForUniversalSegmentation
 
 def load_model_and_processor(model_ckpt: str):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model = Mask2FormerForUniversalSegmentation.from_pretrained(model_ckpt).to(torch.device(device))
     model.eval()
-    image_preprocessor = MaskFormerImageProcessor.from_pretrained(model_ckpt)
+    image_preprocessor = Mask2FormerImageProcessor.from_pretrained(model_ckpt)
     return model, image_preprocessor
 
 def load_default_ckpt(segmentation_task: str):