Spaces:
Paused
Paused
Zhen Ye
commited on
Commit
·
38a60a8
1
Parent(s):
29e1c2b
Update Lite model to use spencercdz/YOLOv8m_defence
Browse files
models/detectors/yolov8.py
CHANGED
|
@@ -12,8 +12,8 @@ from models.detectors.base import DetectionResult, ObjectDetector
|
|
| 12 |
class HuggingFaceYoloV8Detector(ObjectDetector):
|
| 13 |
"""YOLOv8 detector whose weights are fetched from the Hugging Face Hub."""
|
| 14 |
|
| 15 |
-
REPO_ID = "
|
| 16 |
-
WEIGHT_FILE = "
|
| 17 |
supports_batch = True
|
| 18 |
max_batch_size = 8
|
| 19 |
|
|
|
|
| 12 |
class HuggingFaceYoloV8Detector(ObjectDetector):
|
| 13 |
"""YOLOv8 detector whose weights are fetched from the Hugging Face Hub."""
|
| 14 |
|
| 15 |
+
REPO_ID = "spencercdz/YOLOv8m_defence"
|
| 16 |
+
WEIGHT_FILE = "yolov8m_defence.pt"
|
| 17 |
supports_batch = True
|
| 18 |
max_batch_size = 8
|
| 19 |
|