update: added netron , display changes

app.py

@@ -1,57 +1,48 @@
 import gradio as gr
 import os
+import threading
 
+# Paths for images
 yolov5_result = os.path.join(os.getcwd(), "data/xai/yolov5.png")
 yolov8_result = os.path.join(os.getcwd(), "data/xai/yolov8.png")
 yolov5_dff = os.path.join(os.getcwd(), "data/xai/yolov5_dff.png")
 yolov8_dff = os.path.join(os.getcwd(), "data/xai/yolov8_dff.png")
 
-def view_model(selected_models):
-    """Generate Netron visualization for the selected models."""
-    netron_html = ""
-    for model in selected_models:
-        if model=="yolov8s":
-            netron_html = f"""
-            <iframe 
-                src="https://netron.app/?url=https://huggingface.co/spaces/BhumikaMak/NeuralVista/resolve/main/weight_files/yolov8s.pt" 
-                width="100%" 
-                height="800" 
-                frameborder="0">
-            </iframe>
-            """
-        if model == "yolov5":
-            netron_html = f"""
-            <iframe 
-                src="https://netron.app/?url=https://huggingface.co/FFusion/FFusionXL-BASE/blob/main/vae_encoder/model.onnx" 
-                width="100%" 
-                height="800" 
-                frameborder="0">
-            </iframe>
-            """
-    return netron_html if netron_html else "<p>No valid models selected for visualization.</p>"
+# Netron HTML templates
+def get_netron_html(model_url):
+    return f"""
+        <iframe 
+            src="{model_url}" 
+            width="100%" 
+            height="800" 
+            frameborder="0">
+        </iframe>
+    """
+
+# URLs for Netron visualizations
+yolov5_url = "https://netron.app/?url=https://huggingface.co/FFusion/FFusionXL-BASE/blob/main/vae_encoder/model.onnx"
+yolov8_url = "https://netron.app/?url=https://huggingface.co/spaces/BhumikaMak/NeuralVista/resolve/main/weight_files/yolov8s.pt"
 
 custom_css = """
 body {
-    background-color: black; /* Navy blue background */
-    background-size: 1800px 1800px; /* Grid cell size */
-    height: 100%; /* Ensure body height is 100% of the viewport */
-    margin: 0; /* Remove default margin */
-    overflow-y: auto; /* Allow vertical scrolling if needed */
+    background-color: black;
+    background-size: 1800px 1800px;
+    height: 100%;
+    margin: 0;
+    overflow-y: auto;
 }
 #neural-vista-title {
-    color: #800000 !important;  /* Purple color for the title */
-    font-size: 32px;           /* Adjust font size as needed */
+    color: #800000 !important;
+    font-size: 32px;
     font-weight: bold;
     text-align: center;
 }
 #neural-vista-text {
-    color: white  !important;  /* Purple color for the title */
-    font-size: 18px;           /* Adjust font size as needed */
+    color: white  !important;
+    font-size: 18px;
     font-weight: bold;
     text-align: center;
-    
 }
-
 #highlighted-text {
     font-weight: bold;
     color: #1976d2;
@@ -62,17 +53,21 @@ with gr.Blocks(css=custom_css) as demo:
     gr.HTML("""
       <div style="border: 2px solid #a05252; padding: 20px; border-radius: 8px;">
         <span style="color: #E6E6FA; font-family: 'Papyrus', cursive; font-weight: bold; font-size: 32px;">NeuralVista</span><br><br>
-        <span style="color: #E6E6FA; font-family: 'Papyrus', cursive; font-size: 18px;">A harmonious framework of tools ☼ designed to illuminate the inner workings of AI.</span>
+        <span style="color: #E6E6FA; font-family: 'Papyrus', cursive; font-size: 18px;">A harmonious framework of tools <span style="color: yellow; font-family: 'Papyrus', cursive; font-size: 18px;">☼</span> designed to illuminate the inner workings of AI.</span>
       </div>
     """)
+    
     with gr.Row():
         with gr.Column():
             gr.Markdown("Yolov5")
             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.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()
+demo.launch()