Spaces:
Sleeping
Sleeping
Venkat V
commited on
Commit
·
6ea5d07
1
Parent(s):
c2717d6
fixed device call in yolo
Browse files- yolo_module/__init__.py +1 -0
yolo_module/__init__.py
CHANGED
|
@@ -7,6 +7,7 @@ import easyocr
|
|
| 7 |
|
| 8 |
# Load YOLO model
|
| 9 |
MODEL_PATH = "models/best.pt"
|
|
|
|
| 10 |
model = YOLO(MODEL_PATH).to(device)
|
| 11 |
print(f"✅ YOLO model loaded on: {device}")
|
| 12 |
|
|
|
|
| 7 |
|
| 8 |
# Load YOLO model
|
| 9 |
MODEL_PATH = "models/best.pt"
|
| 10 |
+
device = get_device()
|
| 11 |
model = YOLO(MODEL_PATH).to(device)
|
| 12 |
print(f"✅ YOLO model loaded on: {device}")
|
| 13 |
|