Update app.py

app.py

@@ -37,18 +37,22 @@ 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])
-        extracted_tags.append(class_name)  # Only appending the class_name
+        extracted_tags.append(class_name)  # Append only the class name
         draw.text((x, y), class_name, fill='black')
 
-    return image_mask, extracted_tags
+    # Joining all tags into a single string separated by " | "
+    tags_string = " | ".join(extracted_tags)
+
+    return image_mask, tags_string
+