added legend

app.py

@@ -77,15 +77,16 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
             result_display = gr.Image(label="Model prediction", visible=True)
             error_text = gr.Markdown("", visible=False)
             with gr.Row():
-                gr.Markdown("The legend of the classes is the following (format **ID**: **name** - Color: **color**)")
+                gr.Markdown("The legend of the classes is the following (format **name** -> **color**)")
                 string = ""
-                for i, (id, name, color) in enumerate(legendHandling()):
-                    if i and not i % 3:
-                        with gr.Row():
-                            gr.Markdown(string)
-                            string = ""
-                    
-                    string += f"**{id}**: {name} - Color: {color}"
+                with gr.Row():
+                    for i, (id, name, color) in enumerate(legendHandling()):
+                        if i and not i % 2:
+                            with gr.Row():
+                                gr.Markdown(string)
+                                string = ""
+                        
+                        string += f"\t **{name}** -> {color}"
 
     with gr.Row():
         gr.Markdown("## Preloaded images to be used for testing the model")