Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -9,7 +9,7 @@ import supervision as sv
|
|
9 |
from ultralytics import YOLO
|
10 |
|
11 |
app = FastAPI()
|
12 |
-
model = YOLO("models/
|
13 |
|
14 |
|
15 |
def parse_detection(detections):
|
|
|
9 |
from ultralytics import YOLO
|
10 |
|
11 |
app = FastAPI()
|
12 |
+
model = YOLO("models/best_v21.pt", task="detect")
|
13 |
|
14 |
|
15 |
def parse_detection(detections):
|