Update README.md
Browse files
README.md
CHANGED
|
@@ -34,7 +34,8 @@ model = Yolor(
|
|
| 34 |
cfg='yolor/cfg/yolor_p6.cfg',
|
| 35 |
weights='kadirnar/yolor-p6',
|
| 36 |
imgsz=640,
|
| 37 |
-
device='cuda:0'
|
|
|
|
| 38 |
)
|
| 39 |
|
| 40 |
model.classes = None
|
|
|
|
| 34 |
cfg='yolor/cfg/yolor_p6.cfg',
|
| 35 |
weights='kadirnar/yolor-p6',
|
| 36 |
imgsz=640,
|
| 37 |
+
device='cuda:0',
|
| 38 |
+
hf_model=True
|
| 39 |
)
|
| 40 |
|
| 41 |
model.classes = None
|