Update app.py

app.py

@@ -1,12 +1,9 @@
 import gradio as gr
 from huggingface_hub import hf_hub_download
 import torch
+from ultralytics import YOLO
 
-# Download the YOLOv5 model from Hugging Face
-model_path = hf_hub_download(repo_id="mkhug98/EchoYolo", filename="best.onnx")
-
-# Load the YOLOv5 model
-model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path)
+model = YOLO("best.onnx")
 
 # Function to perform object detection
 def detect_objects(image):