test

app.py

@@ -17,25 +17,24 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
-        [204, 87, 92],  # background (Reddish)
-        [112, 185, 212],  # hat (Blue)
-        [196, 160, 122],  # hair (Brown)
-        [106, 135, 242],  # sunglasses (Light Blue)
-        [91, 192, 222],  # upper-clothes (Turquoise)
-        [255, 192, 203],  # skirt (Pink)
-        [176, 224, 230],  # pants (Light Blue)
-        [222, 49, 99],  # dress (Red)
-        [139, 69, 19],  # belt (Brown)
-        [255, 0, 0],  # left-shoe (Red)
-        [0, 0, 255],  # right-shoe (Blue)
-        [255, 228, 181],  # face (Peach)
-        [128, 0, 0],  # left-leg (Maroon)
-        [0, 128, 0],  # right-leg (Green)
-        [255, 99, 71],  # left-arm (Tomato)
-        [0, 255, 0],  # right-arm (Lime)
-        [128, 0, 128],  # bag (Purple)
-        [255, 255, 0]  # scarf (Yellow)
-
+        [204, 87, 92],
+        [112, 185, 212],
+        [45, 189, 106],
+        [234, 123, 67],
+        [78, 56, 123],
+        [210, 32, 89],
+        [90, 180, 56],
+        [155, 102, 200],
+        [33, 147, 176],
+        [255, 183, 76],
+        [67, 123, 89],
+        [190, 60, 45],
+        [134, 112, 200],
+        [56, 45, 189],
+        [200, 56, 123],
+        [87, 92, 204],
+        [120, 56, 123],
+        [45, 78, 123]
     ]
 
 labels_list = []
@@ -102,10 +101,10 @@ def sepia(input_img):
     return fig
 
 demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(shape=[400, 600]),
+                    inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
                     examples=["person-1.jpg", "person-2.jpg", "person-3.jpg", "person-4.jpg", "person-5.jpg"],
                     allow_flagging='never')
 
 
-demo.launch()
+demo.launch()

labels.txt

@@ -1,18 +1,18 @@
-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
+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