update ux interface

app.py

@@ -10,7 +10,7 @@ yolov8_dff = os.path.join(os.getcwd(), "data/xai/yolov8_dff.png")
 # Netron HTML templates
 def get_netron_html(model_url):
     return f"""
-        <div style="background-color: black; padding: 5px; border: 1px solid white;">
+        <div style="background-color: black; padding: 1px; border: 0.5px solid white;">
             <iframe 
                 src="{model_url}" 
                 width="100%" 
@@ -60,15 +60,15 @@ with gr.Blocks(css=custom_css) as demo:
     
     with gr.Row():
         with gr.Column():
-            gr.Markdown("Yolov5")
+            gr.HTML(get_netron_html(yolov5_url))
             gr.Image(yolov5_result, label="Detections & Interpretability Map")
             gr.Image(yolov5_dff, label="Feature Factorization & discovered concept")
-            gr.HTML(get_netron_html(yolov5_url))
+            
 
         with gr.Column():
-            gr.Markdown("Yolov8")
+            gr.HTML(get_netron_html(yolov8_url))
             gr.Image(yolov8_result, label="Detections & Interpretability Map")
             gr.Image(yolov8_dff, label="Feature Factorization & discovered concept")
-            gr.HTML(get_netron_html(yolov8_url))
+            
 
 demo.launch()