modify

app.py

@@ -5,14 +5,14 @@ import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
 import tensorflow as tf
-from transformers import MaskFormerImageProcessor, MaskFormerForInstanceSegmentation
+from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 import requests
 
 
-feature_extractor = MaskFormerImageProcessor.from_pretrained(
+feature_extractor = SegformerFeatureExtractor.from_pretrained(
     "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
-model = MaskFormerForInstanceSegmentation.from_pretrained(
+model = SegformerForSemanticSegmentation.from_pretrained(
     "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )