Spaces:
Sleeping
Sleeping
Initial commit
Browse files- main.py +1 -1
- models/best_v22.pt +3 -0
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_v2.pt", task="detect")
|
13 |
|
14 |
|
15 |
def parse_detection(detections):
|
models/best_v22.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7214823d28af925a1fd4913f3851e8002f4b3725bb3641a79ed473db7c403807
|
3 |
+
size 6254499
|