Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import cv2
|
|
3 |
import torch
|
4 |
|
5 |
# Load your YOLO model
|
6 |
-
model = torch.hub.load('ultralytics/yolov5', 'custom', path='
|
7 |
|
8 |
# Function for YOLO detection
|
9 |
def detect_objects(image):
|
|
|
3 |
import torch
|
4 |
|
5 |
# Load your YOLO model
|
6 |
+
model = torch.hub.load('ultralytics/yolov5', 'custom', path='yolov8_Medium.pt')
|
7 |
|
8 |
# Function for YOLO detection
|
9 |
def detect_objects(image):
|