Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from transformers import pipeline
|
|
| 8 |
from PIL import Image
|
| 9 |
|
| 10 |
# Load YOLOv5 model from torch hub
|
| 11 |
-
yolo_model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
|
| 12 |
yolo_model.conf = 0.4 # confidence threshold
|
| 13 |
yolo_model.classes = [0] # only detect persons (class 0)
|
| 14 |
|
|
|
|
| 8 |
from PIL import Image
|
| 9 |
|
| 10 |
# Load YOLOv5 model from torch hub
|
| 11 |
+
yolo_model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, trust_repo=True)
|
| 12 |
yolo_model.conf = 0.4 # confidence threshold
|
| 13 |
yolo_model.classes = [0] # only detect persons (class 0)
|
| 14 |
|