Mountchicken commited on
Commit
96d7f77
1 Parent(s): 8cecaa1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def arg_parse():
55
  parser.add_argument(
56
  '--device',
57
  type=str,
58
- default='cuda:0',
59
  help='The device used for inference.')
60
  args = parser.parse_args()
61
  return args
 
55
  parser.add_argument(
56
  '--device',
57
  type=str,
58
+ default='cpu',
59
  help='The device used for inference.')
60
  args = parser.parse_args()
61
  return args