revert -1

app.py

@@ -122,25 +122,36 @@ with gr.Blocks(css=custom_css) as interface:
         outputs=sample_display,
     )
 
-    # Parallel execution with separate updates
-    def process_and_update(sample_choice, uploaded_image, selected_models):
-        """Handles parallel processing and Netron updates."""
-        # Start Netron visualization in a separate thread
-        def update_netron():
+    # Multi-threaded processing
+    def run_both(sample_choice, uploaded_image, selected_models):
+        results = []
+        netron_html = ""
+
+        # Thread to process the image
+        def process_thread():
+            nonlocal results
+            results = process_image(sample_choice, uploaded_image, selected_models)
+
+        # Thread to generate Netron visualization
+        def netron_thread():
+            nonlocal netron_html
             netron_html = view_model(selected_models)
-            netron_display.update(netron_html)  # Update Netron asynchronously
 
-        threading.Thread(target=update_netron).start()
+        # Launch threads
+        t1 = threading.Thread(target=process_thread)
+        t2 = threading.Thread(target=netron_thread)
+        t1.start()
+        t2.start()
+        t1.join()
+        t2.join()
 
-        # Process the image in the main thread and return results
-        results = process_image(sample_choice, uploaded_image, selected_models)
-        return results
+        return results, netron_html
 
     # Run button click
     run_button.click(
-        fn=process_and_update,
+        fn=run_both,
         inputs=[sample_selection, upload_image, selected_models],
-        outputs=[result_gallery],
+        outputs=[result_gallery, netron_display],
     )
 
 # Launch Gradio interface