final commit

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)
 
@@ -86,12 +86,12 @@ with gr.Blocks(title="Semantic Segmentation Predictors") as demo:
                         with gr.Column(scale=1):
                             color_box0 = f"""<span style='display:inline-block; width:15px; height:15px; 
                                 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"<div style='display:flex; align-items:center; margin-bottom:-5px;'><b>{legend[i][1]}</b> → {color_box0}</div>")
+                            gr.HTML(f"<div style='display:flex; align-items:center; margin-bottom:-10px; margin-top:-5px;'><b>{legend[i][1]}</b> → {color_box0}</div>")
                         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][3][0]},{legend[i+1][3][1]},{legend[i+1][3][2]}); margin-left:6px; border:1px solid #000;'></span>"""
-                                gr.HTML(f"<div style='display:flex; align-items:center; margin-bottom:-5px;'><b>{legend[i+1][1]}</b> → {color_box1}</div>")
+                                gr.HTML(f"<div style='display:flex; align-items:center; margin-bottom:-10px; margin-top:-5px;'><b>{legend[i+1][1]}</b> → {color_box1}</div>")
                             else:
                                 gr.Markdown("")