SkalskiP commited on
Commit
d54384c
·
1 Parent(s): 3338cd9

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -204,6 +204,8 @@ def track(
204
  "--overwrite",
205
  "--model",
206
  model,
 
 
207
  "--tracker",
208
  tracker,
209
  "--model.confidence",
 
204
  "--overwrite",
205
  "--model",
206
  model,
207
+ "--model.device",
208
+ "cuda",
209
  "--tracker",
210
  tracker,
211
  "--model.confidence",