update: headers

app.py

@@ -323,6 +323,7 @@ with gr.Blocks(css=custom_css, theme="default") as demo:
     
     # Results and visualization
     with gr.Row(elem_classes="custom-row"):
+        gr.Markdown(""" ##Detections & Interpretability Map  """)
         result_gallery = gr.Gallery(
             label="Results",
             rows=1, 
@@ -330,6 +331,7 @@ with gr.Blocks(css=custom_css, theme="default") as demo:
             columns=1 ,
             object_fit="contain"
         ) 
+        gr.Markdown(""" ## Model Architecture""")
         netron_display = gr.HTML(label="Netron Visualization")
 
     # Update sample image
@@ -338,10 +340,10 @@ with gr.Blocks(css=custom_css, theme="default") as demo:
         inputs=sample_selection,
         outputs=sample_display,
     )
-
+    gr.Markdown(""" ## Feature Factorization & discovered concepts. """)
     with gr.Row(elem_classes="custom-row"):
         dff_gallery = gr.Gallery(
-            label="Deep Feature Factorization",
+            label="Feature Factorization & discovered concept",
             rows=2,          # 8 rows
             columns=4,       # 1 image per row
             object_fit="fit",