Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import zipfile
|
|
| 6 |
|
| 7 |
# Load the YOLOv5 model (ensure the path is correct)
|
| 8 |
model = torch.hub.load(
|
| 9 |
-
'ultralytics/yolov5', 'custom', path='best.pt'
|
| 10 |
)
|
| 11 |
|
| 12 |
# Global variables to store images and labels for carousel functionality
|
|
|
|
| 6 |
|
| 7 |
# Load the YOLOv5 model (ensure the path is correct)
|
| 8 |
model = torch.hub.load(
|
| 9 |
+
'ultralytics/yolov5', 'custom', path='best-3.pt'
|
| 10 |
)
|
| 11 |
|
| 12 |
# Global variables to store images and labels for carousel functionality
|