Hugging Face's logo

Spaces:
udayzee05
/
object-counting
Running

App Files Community
chauhuynh90 commited on
Commit
b2f1c9c
·
1 Parent(s): 00b33b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def parse_args(known=False):
40
  parser.add_argument(
41
  "--device",
42
  "-dev",
43
- default="cuda:0",
44
  type=str,
45
  help="cuda or cpu",
46
  )
 
40
  parser.add_argument(
41
  "--device",
42
  "-dev",
43
+ default="cpu",
44
  type=str,
45
  help="cuda or cpu",
46
  )