test run

app.py

@@ -9,7 +9,7 @@ import numpy as np
 from yolov5 import xai_yolov5
 from yolov8 import xai_yolov8s
 import requests
-
+"""
 # Sample images directory
 sample_images = {
     "Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
@@ -86,12 +86,12 @@ def view_netron_model():
 # Custom CSS for styling (optional)
 custom_css = """
 #run_button {
-    background-color: purple;
-    color: white;
-    width: 120px;
-    border-radius: 5px;
-    font-size: 14px;
-}
+   # background-color: purple;
+   # color: white;
+  #  width: 120px;
+  #  border-radius: 5px;
+  #  font-size: 14px;
+#}
 """
 
 # Gradio UI
@@ -157,3 +157,24 @@ with gr.Blocks(css=custom_css) as interface:
 # Launch Gradio app
 if __name__ == "__main__":
     interface.launch(share=True)
+"""
+
+import gradio as gr
+import netron
+
+# Define a function to visualize the model
+def visualize_model(model_file):
+    # Launch the Netron viewer in a web browser
+    netron.start(model_file)
+    return "Model visualization launched in browser."
+
+# Create a Gradio interface
+interface = gr.Interface(
+    fn=visualize_model,
+    inputs=gr.File(label="Upload Model File"),  # Model file input
+    outputs="text",  # Just a text output to indicate launch
+    live=True
+)
+
+interface.launch()
+