new layout

app.py

@@ -76,17 +76,17 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
         with gr.Column():
             result_display = gr.Image(label="Model prediction", visible=True)
             error_text = gr.Markdown("", visible=False)
+            gr.Markdown("The legend of the classes is the following (format **name** -> **color**)")
             with gr.Row():
-                gr.Markdown("The legend of the classes is the following (format **name** -> **color**)")
-                string = ""
-                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}"
+                for i in range(0, len(legendHandling()), 2):
+                    with gr.Row():
+                        gr.Markdown(f"**{legendHandling()[i][1]}** → {legendHandling()[i][2]}")
+
+                        if i + 1 < len(legendHandling()):
+                            gr.Markdown(f"**{legendHandling()[i + 1][1]}** → {legendHandling()[i + 1][2]}")
+                        else:
+                            gr.Markdown("")  # Fills the second column if odd number of items
+
 
     with gr.Row():
         gr.Markdown("## Preloaded images to be used for testing the model")