kbarnard commited on
Commit
79267d6
·
1 Parent(s): a9a0edc

Remove class filter, drop conf threshold

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -7,8 +7,7 @@ model = YOLO(model_path)
7
 
8
 
9
  PREDICT_KWARGS = {
10
- "classes": 0,
11
- "conf": 0.25,
12
  }
13
 
14
 
 
7
 
8
 
9
  PREDICT_KWARGS = {
10
+ "conf": 0.15,
 
11
  }
12
 
13