revert -3

app.py

@@ -1,4 +1,3 @@
-import netron
 import threading
 import gradio as gr
 import os
@@ -7,8 +6,6 @@ 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 = {
@@ -70,6 +67,31 @@ 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")
     
@@ -122,18 +144,13 @@ with gr.Blocks(css=custom_css) as interface:
         outputs=sample_display,
     )
 
-    # Process image and display results, also trigger Netron visualization when run button is clicked
+    # Run both functions concurrently on button click
     run_button.click(
-        fn=lambda sample_choice, uploaded_image, yolo_versions: [ 
-            view_model(yolo_versions),  # Display model visualization
-            process_image(sample_choice, uploaded_image, yolo_versions), 
-
-        ],
+        fn=run_both,
         inputs=[sample_selection, upload_image, selected_models],
         outputs=[result_gallery, netron_display],
     )
 
 # Launching Gradio app
 if __name__ == "__main__":
-    interface.launch(share=True)
- 
+    interface.launch(share=True)