new style

app.py

@@ -78,15 +78,10 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
             error_text = gr.Markdown("", visible=False)
             gr.Markdown("The legend of the classes is the following (format **name** -> **color**)")
             with gr.Row():
-                for i in range(0, len(legendHandling()), 2):
+                legend = legendHandling()
+                for i in range(0, len(legend), 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
-
+                        gr.Markdown(f"**{legend[i][1]}** → {legend[i][2]}  \t\t\t **{legend[i+1][1]}** → {legend[i+1][2]}")
 
     with gr.Row():
         gr.Markdown("## Preloaded images to be used for testing the model")