Update detector/gesture_detector.py
Browse files
detector/gesture_detector.py
CHANGED
|
@@ -4,6 +4,10 @@ import cv2
|
|
| 4 |
import os
|
| 5 |
from datetime import datetime
|
| 6 |
import base64
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
# Load model YOLO hanya sekali
|
| 9 |
model = YOLO("best.pt")
|
|
|
|
| 4 |
import os
|
| 5 |
from datetime import datetime
|
| 6 |
import base64
|
| 7 |
+
from ultralytics.nn.tasks import DetectionModel
|
| 8 |
+
from torch.serialization import add_safe_globals
|
| 9 |
+
|
| 10 |
+
add_safe_globals([DetectionModel])
|
| 11 |
|
| 12 |
# Load model YOLO hanya sekali
|
| 13 |
model = YOLO("best.pt")
|