Update app.py

app.py

@@ -8,10 +8,10 @@ import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "mattmdjaga/segformer_b2_clothes"
+    "nvidia/segformer-b0-finetuned-cityscapes-768-768"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "mattmdjaga/segformer_b2_clothes"
+    "nvidia/segformer-b0-finetuned-cityscapes-768-768"
 )
 
 def ade_palette():
@@ -35,6 +35,7 @@ def ade_palette():
         [189, 78, 47],
         [28, 210, 99],
         [78, 145, 57],
+        [98, 0, 57],
     ]
 
 labels_list = []