fixed function parameters

elyha7 · Mar 2, 2023 · 04bfe34 · 04bfe34
1 parent bee95bd
commit 04bfe34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -12,7 +12,7 @@ from face_detector import YoloDetector
 import numpy as np
 from PIL import Image
 
-model = YoloDetector(target_size=720,gpu=0,min_face=90)
+model = YoloDetector(target_size=720, device="cuda:0", min_face=90)
 orgimg = np.array(Image.open('test_image.jpg'))
 bboxes,points = model.predict(orgimg)
 ```