kadirnar commited on
Commit
a21dec2
1 Parent(s): 51bd321

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ pip install yolov7detect
27
  import yolov7
28
 
29
  # load pretrained or custom model
30
- model = yolov7.load('kadirnar/yolov7-tiny-v0.1')
31
 
32
  # set model parameters
33
  model.conf = 0.25 # NMS confidence threshold
 
27
  import yolov7
28
 
29
  # load pretrained or custom model
30
+ model = yolov7.load('kadirnar/yolov7-tiny-v0.1', hf_model=True)
31
 
32
  # set model parameters
33
  model.conf = 0.25 # NMS confidence threshold