concurrent ui component updates

app.py

@@ -122,36 +122,25 @@ with gr.Blocks(css=custom_css) as interface:
         outputs=sample_display,
     )
 
-    # 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
+    # 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():
             netron_html = view_model(selected_models)
+            netron_display.update(netron_html)  # Update Netron asynchronously
 
-        # Launch threads
-        t1 = threading.Thread(target=process_thread)
-        t2 = threading.Thread(target=netron_thread)
-        t1.start()
-        t2.start()
-        t1.join()
-        t2.join()
+        threading.Thread(target=update_netron).start()
 
-        return results, netron_html
+        # Process the image in the main thread and return results
+        results = process_image(sample_choice, uploaded_image, selected_models)
+        return results
 
     # Run button click
     run_button.click(
-        fn=run_both,
+        fn=process_and_update,
         inputs=[sample_selection, upload_image, selected_models],
-        outputs=[result_gallery, netron_display],
+        outputs=[result_gallery],
     )
 
 # Launch Gradio interface