Use OpenEarthMap semantic class names

app.py

@@ -43,6 +43,17 @@ SEGMENTATION_MODEL_ID = "mfaytin/mask2former-satellite"
 DETECTION_MODEL_ID = "bluelabel/satellite-equipment-detection-yolov8n-vhr10"
 DETECTION_FILENAME = "best.pt"
 OUTPUT_ROOT = Path("/tmp/satellite-vision-toolkit")
+OPEN_EARTH_MAP_LABELS = {
+    0: "background",
+    1: "bareland",
+    2: "grass",
+    3: "pavement",
+    4: "road",
+    5: "tree",
+    6: "water",
+    7: "cropland",
+    8: "building",
+}
 
 
 @lru_cache(maxsize=1)
@@ -55,8 +66,7 @@ def load_segmenter():
         .to(device)
         .eval()
     )
-    id2label = {int(key): str(value) for key, value in model.config.id2label.items()}
-    return processor, model, id2label, device
+    return processor, model, OPEN_EARTH_MAP_LABELS, device
 
 
 @lru_cache(maxsize=1)
@@ -266,4 +276,3 @@ RGB PNG/JPEG/WebP/TIFF images work best. Results are image-space estimates, not
 
 if __name__ == "__main__":
     demo.queue(default_concurrency_limit=2).launch()
-