Update app.py

app.py

@@ -1,14 +1,10 @@
 import gradio as gr
-import netron
 import os
-import threading
-import time    
 from PIL import Image
 import cv2
 import numpy as np
 from yolov5 import xai_yolov5
 from yolov8 import xai_yolov8s
-import requests
 
 # Sample images directory
 sample_images = {
@@ -16,9 +12,6 @@ sample_images = {
     "Sample 2": os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
 }
 
-# Preloaded model file path
-preloaded_model_file = os.path.join(os.getcwd(), "weight_files/yolov5.onnx")  # Example path
-
 # Function to load sample image
 def load_sample_image(sample_name):
     image_path = sample_images.get(sample_name)
@@ -49,40 +42,6 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
 
     return result_images
 
-# Start Netron backend
-def start_netron_backend(model_file):
-    def serve_netron():
-        netron.start(model_file, address=("0.0.0.0", 8080), browse=False)
-        #netron.start(model_file, address="0.0.0.0:8080", browse=False)  # Updated Netron arguments
-
-    # Launch Netron in a separate thread
-    threading.Thread(target=serve_netron, daemon=True).start()
-
-    # Wait until Netron server is ready
-    def wait_for_netron(url, timeout=10):
-        start_time = time.time()
-        while time.time() - start_time < timeout:
-            try:
-                response = requests.get(url)
-                if response.status_code == 200:
-                    return True
-            except requests.ConnectionError:
-                time.sleep(0.5)
-        return False
-
-    # Check server readiness
-    wait_for_netron("http://localhost:8080/", timeout=15)
-
-# View Netron model
-def view_netron_model():
-    # Ensure model exists
-    if not os.path.exists(preloaded_model_file):
-        return "Model file not found."
-
-    # Start Netron backend
-    start_netron_backend(preloaded_model_file)
-    return gr.HTML('<iframe src="http://localhost:8080/" width="100%" height="600px"></iframe>')
-
 # Custom CSS for styling (optional)
 custom_css = """
  run_button {
@@ -128,7 +87,7 @@ with gr.Blocks(css=custom_css) as interface:
                 label="Selected Sample Image",
             )
 
-    # Results and Netron
+    # Results
     with gr.Row():
         result_gallery = gr.Gallery(
             label="Results",
@@ -137,9 +96,6 @@ with gr.Blocks(css=custom_css) as interface:
             height=500,
         )
 
-        # Display Netron iframe
-        netron_display = gr.HTML(view_netron_model())
-
     # Sample selection update
     sample_selection.change(
         fn=load_sample_image,