Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ DETECTION_CLASSES = ["Longitudinal", "Pothole", "Transverse"]
|
|
| 40 |
MAX_IMAGES = 500
|
| 41 |
|
| 42 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 43 |
-
model = YOLO('./data/
|
| 44 |
if device == "cuda":
|
| 45 |
model.half()
|
| 46 |
|
|
|
|
| 40 |
MAX_IMAGES = 500
|
| 41 |
|
| 42 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 43 |
+
model = YOLO('./data/yolo11n (2).pt').to(device)
|
| 44 |
if device == "cuda":
|
| 45 |
model.half()
|
| 46 |
|