Update detect_faces.py
Browse files- 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 |
|