new check

app.py

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