Update app.py

app.py

@@ -103,11 +103,20 @@ def predict(input_image):
 
     return image_pred, image_gt, gt_visibility
 
+# Generate the color legend markdown dynamically
+legend_markdown = "### Color Legend\n\n"
+for name, color in zip(class_names, colors):
+    color_rgb = f'rgb({color[0]},{color[1]},{color[2]})'
+    legend_markdown += f'- <span style="color: {color_rgb};">&#9608;</span> **{name}**\n'
+
 # Create Gradio interface
 with gr.Blocks() as interface:
-    gr.Markdown("# Advanced Detection of Cochlear Hair Cells Using YOLOv10 in Auditory Diagnostics")
+    gr.Markdown("## Advanced Detection of Cochlear Hair Cells Using YOLOv10 in Auditory Diagnostics")
     gr.Markdown("Select an example or upload an image to see predictions.")
 
+    # Add the color legend
+    gr.Markdown(legend_markdown)
+
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(type="pil", label="Input Image")