Update app.py

app.py

@@ -76,8 +76,9 @@ def draw_plot(pred_img, seg):
 
 def segmentation_to_text(seg):
     seg = seg.numpy()
-    labels_text = [labels_list[label] for label in np.unique(seg)]
-    return ", ".join(labels_text)
+    unique_labels, counts = np.unique(seg, return_counts=True)
+    labels_text = [f"{labels_list[label]}: {count}" for label, count in zip(unique_labels, counts)]
+    return "\n".join(labels_text)
 
 def sepia(input_img):
     input_img = Image.fromarray(input_img)