test endpoint deployment netron

app.py

@@ -48,7 +48,7 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
 
 def start_netron_backend(model_file):
     def serve_netron():
-        netron.start(model_file, port=8080)  # Start Netron on port 8080
+        netron.start(model_file, host='0.0.0.0', port=8080)  # Start Netron on port 8080
     
     # Launch Netron in a separate thread
     threading.Thread(target=serve_netron, daemon=True).start()
@@ -61,8 +61,7 @@ def view_netron_model():
 
     # Start Netron backend
     start_netron_backend(preloaded_model_file)
-    return '<iframe src="http://localhost:8080/" width="100%" height="600px"></iframe>'
-
+    return gr.HTML(f'<iframe src="http://localhost:8080/" width="100%" height="600px"></iframe>')
 
 # Custom CSS for styling (optional)
 custom_css = """
@@ -118,7 +117,7 @@ with gr.Blocks(css=custom_css) as interface:
             height=500,
         )
 
-        netron_display = gr.HTML(label="Model Visualization")
+        # netron_display = gr.HTML(label="Model Visualization")
 
     # Sample selection update
     sample_selection.change(
@@ -135,7 +134,8 @@ with gr.Blocks(css=custom_css) as interface:
     )
 
     # Netron model visualization
-    netron_display.value = view_netron_model()
+    interface.add_component(gr.HTML(view_netron_model))
+    # netron_display.value = view_netron_model()
 
 # Launch Gradio app
 if __name__ == "__main__":