Spaces:
Runtime error
Runtime error
fix: change default device to cpu
Browse files
app.py
CHANGED
@@ -82,7 +82,7 @@ def main():
|
|
82 |
"--detector", type=str, default="yolov3", choices=["yolov3", "faster-rcnn"]
|
83 |
)
|
84 |
parser.add_argument(
|
85 |
-
"--device", type=str, default="
|
86 |
)
|
87 |
parser.add_argument("--face-score-threshold", type=float, default=0.5)
|
88 |
parser.add_argument("--landmark-score-threshold", type=float, default=0.3)
|
|
|
82 |
"--detector", type=str, default="yolov3", choices=["yolov3", "faster-rcnn"]
|
83 |
)
|
84 |
parser.add_argument(
|
85 |
+
"--device", type=str, default="cpu", choices=["cuda:0", "cpu"]
|
86 |
)
|
87 |
parser.add_argument("--face-score-threshold", type=float, default=0.5)
|
88 |
parser.add_argument("--landmark-score-threshold", type=float, default=0.3)
|