Spaces:

LeeRuben
/

rae

Sleeping

LeeRuben commited on Nov 5, 2023

Commit

8a362aa

1 Parent(s): 2dc91cb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,9 +17,9 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
-        [204, 87, 92],
-        [112, 185, 212],
-        [45, 189, 106],
         [234, 123, 67],
         [78, 56, 123],
         [210, 32, 89],

 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
+        [255, 0, 0],
+        [0, 255, 0],
+        [0, 0, 255],
         [234, 123, 67],
         [78, 56, 123],
         [210, 32, 89],