Spaces:
Sleeping
Sleeping
Mountchicken
commited on
Commit
•
96d7f77
1
Parent(s):
8cecaa1
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def arg_parse():
|
|
55 |
parser.add_argument(
|
56 |
'--device',
|
57 |
type=str,
|
58 |
-
default='
|
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
|