Update app.py

app.py

@@ -28,16 +28,20 @@ def draw_panoptic_segmentation(segmentation, segments_info):
     instances_counter = defaultdict(int)
     handles = []
     # for each segment, draw its legend
+    # for segment in segments_info:
+    #     segment_id = segment['id']
+    #     segment_label_id = segment['label_id']
+    #     segment_label = model.config.id2label[segment_label_id]
+    #     label = f"{segment_label}-{instances_counter[segment_label_id]}"
+    #     instances_counter[segment_label_id] += 1
+    #     color = viridis(segment_id)
+    #     handles.append(mpatches.Patch(color=color, label=label))
+        
+    # ax.legend(handles=handles)
+
     for segment in segments_info:
         segment_id = segment['id']
-        segment_label_id = segment['label_id']
-        segment_label = model.config.id2label[segment_label_id]
-        label = f"{segment_label}-{instances_counter[segment_label_id]}"
-        instances_counter[segment_label_id] += 1
-        color = viridis(segment_id)
-        handles.append(mpatches.Patch(color=color, label=label))
-        
-    ax.legend(handles=handles)
+        color = viridis(segment_id
     
     # Save the figure to a buffer and convert it to a PIL image
     buf = BytesIO()