Update app.py

app.py

@@ -8,10 +8,10 @@ import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "nvidia/segformer-b5-finetuned-ade-640-640"
+    "nvidia/segformer-b2-finetuned-ade-512-512"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "nvidia/segformer-b5-finetuned-ade-640-640"
+    "nvidia/segformer-b2-finetuned-ade-512-512"
 )
 
 def ade_palette():
@@ -235,7 +235,7 @@ def sepia(input_img):
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
-                    examples=["ADE_val_00000001.jpeg", "ADE_val_00001159.jpg", "ADE_val_00001248.jpg", "ADE_val_00001472.jpg"],
+                    examples=["kitchen.jpg", "bridge.jpg", "red.jpg", "livingroom.jpg"],
                     allow_flagging='never')