Update app.py

app.py

@@ -158,21 +158,22 @@ with gr.Blocks(css=custom_css) as interface:
 if __name__ == "__main__":
     interface.launch(share=True)
 """
+
 import gradio as gr
 import torch
 from ultralytics import YOLO
 import os
 
-def visualize_yolov5(model_path):
+# Assuming the YOLOv5 model is located in the same directory as this script
+model_path = os.path.join(os.getcwd(), "weight_files/yolov5.onnx")   # Replace with the actual model directory name
+
+def visualize_yolov5():
   """
   Visualizes the given YOLOv5 model using Netron.
-
-  Args:
-    model_path: The path to the YOLOv5 model directory.
   """
   try:
     # Load the YOLOv5 model
-    model = YOLO(model_path)
+    model = YOLO(model_path) 
 
     # Extract the PyTorch model 
     pytorch_model = model.model
@@ -190,10 +191,10 @@ def visualize_yolov5(model_path):
 # Create the Gradio interface
 iface = gr.Interface(
     fn=visualize_yolov5,
-    inputs="text",
+    inputs=None,  # No input required
     outputs="text", 
     title="Netron YOLOv5 Model Visualization",
-    description="Enter the path to your YOLOv5 model directory."
+    description="Visualize the YOLOv5 model."
 )
 
 # Launch the Gradio app