app.py

@@ -80,10 +80,15 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
             with gr.Row():
                 legend = legendHandling()
                 for i in range(0, len(legend), 2):
-                        if i + 1 < len(legend):
-                            gr.Markdown(f"**{legend[i][1]}** → {legend[i][2]}  \t\t\t **{legend[i+1][1]}** → {legend[i+1][2]}")
-                        else:
+                    with gr.Row():
+                        with gr.Column(scale=1):
                             gr.Markdown(f"**{legend[i][1]}** → {legend[i][2]}")
+                        with gr.Column(scale=1):
+                            if i + 1 < len(legend):
+                                gr.Markdown(f"**{legend[i+1][1]}** → {legend[i+1][2]}")
+                            else:
+                                gr.Markdown("")  # Keeps spacing consistent if list is odd
+
 
     with gr.Row():
         gr.Markdown("## Preloaded images to be used for testing the model")