Update app.py

app.py

@@ -622,7 +622,7 @@ def infer_seg(media, query):
     dyn_color_map_light = {}
     for i, tag in enumerate(unique_tags):
         c = colors[i % len(colors)]
-        dyn_color_map[tag] = f'#{int(c[0]):02x}{int(c[1]):02x}{int(c[2]):02x}'
+        dyn_color_map[entity_names[i]] = f'#{int(c[0]):02x}{int(c[1]):02x}{int(c[2]):02x}'
         dyn_color_map_light[tag] = f'#{int(c[0] * 127.5 + 127.5):02x}{int(c[1] * 127.5 + 127.5):02x}{int(c[2] * 127.5 + 127.5):02x}'
 
     # return answer, masks, path