added color

app.py

@@ -12,12 +12,12 @@ turin_image_dir = "./preloadedImages/turin"
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 
-MODELS = {
-    "BISENET-BASE": loadModel('bisenet', device, 'weight_Base.pth'),
-    "BISENET-BEST": loadModel('bisenet', device, 'weight_Best.pth'),
-    "BISENETV2-BASE": loadModel('bisenetv2', device, 'weight_Base.pth'),
-    "BISENETV2-BEST": loadModel('bisenetv2', device, 'weight_Best.pth')
-}
+# MODELS = {
+#     "BISENET-BASE": loadModel('bisenet', device, 'weight_Base.pth'),
+#     "BISENET-BEST": loadModel('bisenet', device, 'weight_Best.pth'),
+#     "BISENETV2-BASE": loadModel('bisenetv2', device, 'weight_Base.pth'),
+#     "BISENETV2-BEST": loadModel('bisenetv2', device, 'weight_Best.pth')
+# }
 
 image_list = loadPreloadedImages(gta_image_dir, city_image_dir, turin_image_dir)
 
@@ -85,15 +85,15 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
                     with gr.Row():
                         with gr.Column(scale=1):
                             color_box0 = f"""<span style='display:inline-block; width:15px; height:15px; 
-                                background-color:rgb({legend[i][1][0]},{legend[i][1][1]},{legend[i][1][2]}); margin-left:6px; border:1px solid #000;'></span>"""
-                            gr.HTML(f"<div><b>{legend[i][1]}</b> → {legend[i][2]} {color_box0}</div>")
+                                background-color:rgb({legend[i][3][0]},{legend[i][3][1]},{legend[i][3][2]}); margin-left:6px; border:1px solid #000;'></span>"""
+                            gr.HTML(f"<b>{legend[i][1]}</b> → {legend[i][2]} {color_box0}")
                         with gr.Column(scale=1):
                             if i + 1 < len(legend):
                                 color_box1 = f"""<span style='display:inline-block; width:15px; height:15px; 
-                                    background-color:rgb({legend[i+1][1][0]},{legend[i+1][1][1]},{legend[i+1][1][2]}); margin-left:6px; border:1px solid #000;'></span>"""
-                                gr.HTML(f"<div><b>{legend[i+1][1]}</b> → {legend[i+1][2]} {color_box1}</div>")
+                                    background-color:rgb({legend[i+1][3][0]},{legend[i+1][3][1]},{legend[i+1][3][2]}); margin-left:6px; border:1px solid #000;'></span>"""
+                                gr.HTML(f"<b>{legend[i+1][1]}</b> → {legend[i+1][2]} {color_box1}")
                             else:
-                                gr.HTML("")  # Keeps spacing consistent if list is odd
+                                gr.Markdown("")  # Keeps spacing consistent if list is odd
 
 
     with gr.Row():

prova.html

@@ -0,0 +1,3 @@
+<span style='display:inline-block; width:15px; height:15px;
+                             background-color:rgb(255,0,0); margin-left:6px;
+                             border:1px solid #000;'></span>