Update app.py

app.py

@@ -159,13 +159,14 @@ if __name__ == "__main__":
     interface.launch(share=True)
 """
 
+
 import gradio as gr
 import torch
 from ultralytics import YOLO
 import os
 
 # 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
+model_path = os.path.join(os.getcwd(), "weight_files/yolov5.onnx")  # Replace with the actual model directory name
 
 def visualize_yolov5():
   """
@@ -176,7 +177,7 @@ def visualize_yolov5():
     model = YOLO(model_path) 
 
     # Extract the PyTorch model 
-    pytorch_model = model.model
+    pytorch_model = model.model 
 
     # Save the PyTorch model to a temporary file
     temp_model_path = "temp_model" 
@@ -197,5 +198,5 @@ iface = gr.Interface(
     description="Visualize the YOLOv5 model."
 )
 
-# Launch the Gradio app
+# Launch the Gradio app with public link
 iface.launch(share=True)