Update utils/video_processing.py
Browse files
utils/video_processing.py
CHANGED
|
@@ -8,6 +8,10 @@ import torch
|
|
| 8 |
# Path to the YOLO model
|
| 9 |
MODEL_PATH = 'models/yolov8_model.pt'
|
| 10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
# Check if model file exists
|
| 12 |
if not os.path.exists(MODEL_PATH):
|
| 13 |
raise FileNotFoundError(
|
|
|
|
| 8 |
# Path to the YOLO model
|
| 9 |
MODEL_PATH = 'models/yolov8_model.pt'
|
| 10 |
|
| 11 |
+
from ultralytics import YOLO
|
| 12 |
+
model = YOLO(MODEL_PATH)
|
| 13 |
+
print("Model loaded successfully")
|
| 14 |
+
|
| 15 |
# Check if model file exists
|
| 16 |
if not os.path.exists(MODEL_PATH):
|
| 17 |
raise FileNotFoundError(
|