Update app.py

app.py

@@ -37,27 +37,18 @@ def visualize_mask(predicted_semantic_map, class_ids, class_colors):
     return image_mask
 
 def get_out_image(image, predicted_semantic_map):
-    class_centers = get_class_centers(predicted_semantic_map, class_dict)
+    class_centers = get_class_centers(predicted_semantic_map, class_dict) 
     mask = visualize_mask(predicted_semantic_map, class_ids, class_colors)
     image_mask = combine_ims(image, mask, val=128)
     draw = ImageDraw.Draw(image_mask)
-
+    
     extracted_tags = []
     for id, (y, x) in class_centers.items():
         class_name = str(class_names[id - 1])
-        color = class_colors[id - 1]
-        color_hex = "#{:02x}{:02x}{:02x}".format(*color)  # Convert RGB to hex
-
-        symbol = "●"  # You can choose any symbol you like
-        tag_info = f"{symbol} [{color_hex}] {class_name}"
-        extracted_tags.append(tag_info)
+        extracted_tags.append(class_name)  # Only appending the class_name
         draw.text((x, y), class_name, fill='black')
 
-    # Joining all tags into a single string, each tag on a new line
-    tags_string = "\n".join(extracted_tags)
-
-    return image_mask, tags_string
-
+    return image_mask, extracted_tags