Update app.py

app.py

@@ -136,7 +136,8 @@ from PIL import Image
 import numpy as np
 
 # Load the YOLO model
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='https://huggingface.co/ayoubsa/yolo_model/resolve/main/best.pt')  # Replace with your uploaded model's path
+MODEL_URL= 'https://huggingface.co/ayoubsa/yolo_model/resolve/main/best.pt'
+model = YOLO(MODEL_URL)
 
 # Define the prediction function
 def predict(image):