Sa-m commited on
Commit
e35b449
1 Parent(s): 1805721

Update app.py

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