Update README.md
Browse files
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
|