initial netron template

app.py

@@ -158,46 +158,22 @@ with gr.Blocks(css=custom_css) as interface:
 if __name__ == "__main__":
     interface.launch(share=True)
 """
-import gradio as gr
-import os
-import subprocess
-
-# Function to visualize the model
-def visualize_model(model_file):
-    # Create the output directory if it doesn't exist
-    output_dir = "./models/tmp/gradio/"
-    os.makedirs(output_dir, exist_ok=True)  # Create directories if needed
-
-    # Define the file path to save
-    file_path = os.path.join(output_dir, model_file.name)
-
-    # Save the uploaded file
-    with open(file_path, "wb") as f:
-        with open(model_file.name, "rb") as uploaded_file:
-            f.write(uploaded_file.read())  # FIXED: Properly read and write file data
-
-    # Start Netron server to visualize the model
-    port = 8081
-    subprocess.Popen(
-        ["netron", "--host", "0.0.0.0", "--port", str(port), file_path],
-        stdout=subprocess.PIPE,
-        stderr=subprocess.PIPE,
-    )
 
-    # Return URL for visualization
-    netron_url = f"http://localhost:{port}"
-    return f"Model uploaded successfully! View it at: {netron_url}"
+import gradio as gr
+import netron
 
+# Function to visualize the model using Netron
+def visualize_model(model_path):
+    netron.start(model_path)
 
-# Gradio Interface
-demo = gr.Interface(
+# Create Gradio interface
+iface = gr.Interface(
     fn=visualize_model,
-    inputs=gr.File(label="Upload your model file (ONNX, TensorFlow, etc.)"),
-    outputs=gr.Textbox(label="Netron Link"),
-    title="Model Visualizer",
-    description="Upload a model file to visualize its architecture using Netron.",
+    inputs="text",
+    outputs="text",
+    title="Model Visualization with Netron",
+    description="Enter the path to your model file (e.g., .onnx, .h5) to visualize."
 )
 
 # Launch the app
-demo.launch(server_name="localhost", server_port=7860, share = True)
-
+iface.launch(share=True)