HubHop commited on
Commit
5ae2b1c
·
1 Parent(s): 1987539
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def main():
25
  # parser.add_argument('--video', help='Video file or webcam id')
26
 
27
  parser.add_argument(
28
- '--device', default='cpu', help='Device used for inference')
29
  parser.add_argument(
30
  '--palette',
31
  default='cityscapes',
 
25
  # parser.add_argument('--video', help='Video file or webcam id')
26
 
27
  parser.add_argument(
28
+ '--device', default='cuda:0', help='Device used for inference')
29
  parser.add_argument(
30
  '--palette',
31
  default='cityscapes',