local model upload

app.py

@@ -158,22 +158,25 @@ with gr.Blocks(css=custom_css) as interface:
 if __name__ == "__main__":
     interface.launch(share=True)
 """
-
 import gradio as gr
 import netron
 
 # Function to visualize the model using Netron
 def visualize_model(model_path):
-    netron.start(model_path)
+    # Start Netron to visualize the model
+    netron.start(model_path, browse=False)  # Set browse=False to avoid opening a new browser tab
+
+# Specify the local path to your model file
+model_path = os.path.join(os.getcwd(), "weight_files/yolov5.onnx")   # Change this to your actual model file path
 
 # Create Gradio interface
 iface = gr.Interface(
-    fn=visualize_model,
-    inputs="text",
-    outputs="text",
+    fn=lambda: visualize_model(model_path),  # Call the function without user input
+    inputs=None,  # No input required from user
+    outputs="text",  # Output type is text (or you can use "html" if needed)
     title="Model Visualization with Netron",
-    description="Enter the path to your model file (e.g., .onnx, .h5) to visualize."
+    description="This app visualizes a model using Netron."
 )
 
 # Launch the app
-iface.launch(share=True)
+iface.launch()