modify

app.py

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