Sa-m commited on
Commit
6b5b43a
·
1 Parent(s): 64c398d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -14,6 +14,8 @@ def load_model():
14
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
15
  model.conf = 0.38
16
  model.hide_conf=True
 
 
17
  return model
18
 
19
  #, force_reload=True
 
14
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
15
  model.conf = 0.38
16
  model.hide_conf=True
17
+ model.dnn=True
18
+ model.line_thickness=6
19
  return model
20
 
21
  #, force_reload=True