change model

app.py

@@ -7,8 +7,12 @@ from PIL import Image
 import tensorflow as tf
 from transformers import SegformerFeatureExtractor, SegformerForSemanticSegmentation
 
-feature_extractor = SegformerFeatureExtractor.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024")
-model = SegformerForSemanticSegmentation.from_pretrained("nvidia/segformer-b3-finetuned-cityscapes-1024-1024")
+feature_extractor = SegformerFeatureExtractor.from_pretrained(
+    "nvidia/segformer-b3-finetuned-cityscapes-1024-1024"
+)
+model = SegformerForSemanticSegmentation.from_pretrained(
+    "nvidia/segformer-b3-finetuned-cityscapes-1024-1024"
+)
 
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""