yuna0x0 commited on
Commit
d5dda80
1 Parent(s): 34383d4

fix: change default device to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="cuda:0", 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)
 
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)