jovian commited on
Commit
ae429f8
·
1 Parent(s): ee2d685

Add application file

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -38,7 +38,8 @@ class Detection:
38
  postprocess_type='NMS',
39
  postprocess_match_metric='IOU',
40
  postprocess_match_threshold=0.1,
41
- postprocess_class_agnostic=True
 
42
  )
43
 
44
  # Retrieve COCO annotations
 
38
  postprocess_type='NMS',
39
  postprocess_match_metric='IOU',
40
  postprocess_match_threshold=0.1,
41
+ postprocess_class_agnostic=True,
42
+ device=device
43
  )
44
 
45
  # Retrieve COCO annotations