Update app.py

app.py

@@ -8,7 +8,7 @@ dataset = load_dataset("dwb2023/brain-tumor-image-dataset-semantic-segmentation"
 # print(f"Dataset loaded successfully. Number of images: {len(dataset)}")
 
 CATEGORY_COLORS = {
-    1: "red",
+    1: "blue",
     2: "green"
 }
 
@@ -30,13 +30,13 @@ def draw_annotations(index):
         # Draw bounding box with color based on category
         bbox = record["bbox"]
         category_id = record["category_id"]
-        box_color = CATEGORY_COLORS.get(category_id, "blue")  # Default to blue if category not in mapping
+        box_color = CATEGORY_COLORS.get(category_id, "yellow")  # Default to blue if category not in mapping
         draw.rectangle([bbox[0], bbox[1], bbox[0] + bbox[2], bbox[1] + bbox[3]], outline=box_color, width=2)
 
         # Draw segmentation mask
         segmentation = record["segmentation"]
         for seg in segmentation:
-            draw.polygon(seg, outline="blue", width=2)
+            draw.polygon(seg, outline="orange", width=2)
 
         # Prepare additional information
         area = record["area"]