benstaf commited on
Commit
4441043
·
verified ·
1 Parent(s): 64dd02b

Update detect_faces.py

Browse files
Files changed (1) hide show
  1. detect_faces.py +2 -2
detect_faces.py CHANGED
@@ -19,8 +19,8 @@ def check_and_download_model():
19
  f.write(response.content)
20
  return MODEL_PATH
21
 
22
-
23
- model = YOLO(check_and_download_model())
24
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
25
  model.to(device)
26
 
 
19
  f.write(response.content)
20
  return MODEL_PATH
21
 
22
+ model = YOLO("yolov8n.pt")
23
+ #model = YOLO(check_and_download_model())
24
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
25
  model.to(device)
26