Spaces:
Runtime error
Runtime error
HubHop
commited on
Commit
·
5ae2b1c
1
Parent(s):
1987539
update
Browse files
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='
|
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',
|