Spaces:
Build error
Build error
updated main.py
Browse files
main.py
CHANGED
|
@@ -9,7 +9,7 @@ from datetime import timedelta
|
|
| 9 |
VIDEOS_DIR = '.'
|
| 10 |
video_path = os.path.join(VIDEOS_DIR, 'sample_video.mp4')
|
| 11 |
output_json_path = 'output.json'
|
| 12 |
-
model_path = os.path.join('.', '
|
| 13 |
|
| 14 |
# Load YOLOv8 model
|
| 15 |
model = YOLO(model_path) # Load a custom model
|
|
|
|
| 9 |
VIDEOS_DIR = '.'
|
| 10 |
video_path = os.path.join(VIDEOS_DIR, 'sample_video.mp4')
|
| 11 |
output_json_path = 'output.json'
|
| 12 |
+
model_path = os.path.join('.', 'best.pt')
|
| 13 |
|
| 14 |
# Load YOLOv8 model
|
| 15 |
model = YOLO(model_path) # Load a custom model
|