Update app.py

app.py

@@ -1,3 +1,4 @@
+import netron
 import threading
 import gradio as gr
 import os
@@ -6,6 +7,8 @@ import cv2
 import numpy as np
 from yolov5 import xai_yolov5
 from yolov8 import xai_yolov8s
+import time
+import tempfile
 
 # Sample images directory
 sample_images = {
@@ -42,9 +45,9 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
     return result_images
 
 def view_model(selected_models):
-    """Generate Netron visualization for the selected models."""
-    for model in selected_models:
+    for model in selected_models[0]:
         if model == "yolov5":
+            # Embed the Netron viewer using an iframe with the generated URL
             iframe_html = f"""
             <iframe 
                 src="https://netron.app/?url=https://huggingface.co/FFusion/FFusionXL-BASE/blob/main/vae_encoder/model.onnx" 
@@ -67,31 +70,6 @@ custom_css = """
 }
 """
 
-def run_both(sample_choice, uploaded_image, yolo_versions):
-    """Run both image processing and model visualization simultaneously."""
-    results = []
-
-    def process_thread():
-        result_images = process_image(sample_choice, uploaded_image, yolo_versions)
-        results.append(result_images)
-    
-    def model_thread():
-        model_html = view_model(yolo_versions)
-        results.append(model_html)
-
-    # Create threads to run both functions simultaneously
-    process_thread_obj = threading.Thread(target=process_thread)
-    model_thread_obj = threading.Thread(target=model_thread)
-
-    process_thread_obj.start()
-    model_thread_obj.start()
-
-    # Wait for both threads to finish
-    process_thread_obj.join()
-    model_thread_obj.join()
-
-    return results[0], results[1]  # Return processed image results and model visualization
-
 with gr.Blocks(css=custom_css) as interface:
     gr.Markdown("# NeuralVista: Visualize Object Detection of Your Models")
     
@@ -137,20 +115,24 @@ with gr.Blocks(css=custom_css) as interface:
 
         netron_display = gr.HTML(label="Netron Visualization")
 
-    # Update the sample image when the sample is changed
     sample_selection.change(
         fn=load_sample_image,
         inputs=sample_selection,
         outputs=sample_display,
     )
 
-    # Run both functions concurrently on button click
     run_button.click(
-        fn=run_both,
+        fn=process_image,
         inputs=[sample_selection, upload_image, selected_models],
-        outputs=[result_gallery, netron_display],
+        outputs=[result_gallery],
+    )
+
+    selected_models.change(
+        fn=view_model,
+        inputs=selected_models,
+        outputs=netron_display,
     )
 
-# Launching Gradio app
+# Launching Gradio app and handling Netron visualization separately.
 if __name__ == "__main__":
-    interface.launch(share=True)
+    interface.launch(share=True)