vu0018 commited on
Commit
4f6f188
·
verified ·
1 Parent(s): 005cf77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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