Upload 7 files

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-1024-1024"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "mattmdjaga/segformer_b2_clothes"
+    "nvidia/segformer-b0-finetuned-cityscapes-1024-1024"
 )
 
 def ade_palette():
@@ -36,6 +36,7 @@ def ade_palette():
         [120, 56, 123],
         [45, 78, 123],
         [156, 200, 56],
+        [156, 54, 212],
     ]
 
 labels_list = []
@@ -104,7 +105,7 @@ def sepia(input_img):
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
-                    examples=["person-1.jpg", "person-2.jpg", "person-3.jpg", "person-4.jpg","person-5.jpg"],
+                    examples=["street-1.jpg", "street-2.jpg", "street-3.jpg", "street-4.jpg","street-5.webp"],
                     allow_flagging='never')
 
 

labels.txt

@@ -1,18 +1,19 @@
-Background
-Hat
-Hair
-Sunglasses
-Upper-clothes
-Skirt
-Pants
-Dress
-Belt
-Left-shoe
-Right-shoe
-Face
-Left-leg
-Right-leg
-Left-arm
-Right-arm
-Bag
-Scarf
+road
+Sidewalk
+Building
+Wall
+Fence
+Pole
+Traffic light
+Traffic sign
+Vegetation
+Terrain
+Sky
+Person
+Rider
+Car
+Truck
+Bus
+Train
+Motorcycle
+Bicycle