Spaces:

Zengyf-CVer
/

gradio_yolov5_det

Runtime error

Zengyf-CVer commited on Apr 14, 2022

Commit

9e12f9c

1 Parent(s): 400d72b

model loading update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,7 @@ def parse_args(known=False):
     parser.add_argument(
         "--device",
         "-dev",
-        default="0",
         type=str,
         help="cuda or cpu",
     )

     parser.add_argument(
         "--device",
         "-dev",
+        default="cpu",
         type=str,
         help="cuda or cpu",
     )