Update app.py

app.py

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