revert ca043a56b14a1faea9f85a77593e24012c6c677e

app.py

@@ -45,9 +45,9 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
     return result_images
 
 def view_model(selected_models):
-    for model in selected_models[0]:
+    """Generate Netron visualization for the selected models."""
+    for model in selected_models:
         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" 
@@ -115,24 +115,23 @@ 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,
     )
 
+    # Process image and display results, also trigger Netron visualization when run button is clicked
     run_button.click(
-        fn=process_image,
+        fn=lambda sample_choice, uploaded_image, yolo_versions: [
+            process_image(sample_choice, uploaded_image, yolo_versions),  # Process image
+            view_model(yolo_versions)  # Display model visualization
+        ],
         inputs=[sample_selection, upload_image, selected_models],
-        outputs=[result_gallery],
+        outputs=[result_gallery, netron_display],
     )
 
-    selected_models.change(
-        fn=view_model,
-        inputs=selected_models,
-        outputs=netron_display,
-    )
-
-# Launching Gradio app and handling Netron visualization separately.
+# Launching Gradio app
 if __name__ == "__main__":
-    interface.launch(share=True)
+    interface.launch(share=True)