Zengyf-CVer commited on
Commit
3b3708e
1 Parent(s): 06ef385
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -247,7 +247,7 @@ def main(args):
247
  ],
248
  [
249
  "./img_example/Millenial-at-work.jpg",
250
- "0",
251
  "yolov5l",
252
  0.5,
253
  0.45,
@@ -256,7 +256,7 @@ def main(args):
256
  ],
257
  [
258
  "./img_example/zidane.jpg",
259
- "0",
260
  "yolov5m",
261
  0.25,
262
  0.5,
 
247
  ],
248
  [
249
  "./img_example/Millenial-at-work.jpg",
250
+ "cpu",
251
  "yolov5l",
252
  0.5,
253
  0.45,
 
256
  ],
257
  [
258
  "./img_example/zidane.jpg",
259
+ "cpu",
260
  "yolov5m",
261
  0.25,
262
  0.5,