Sa-m commited on
Commit
d5809d1
1 Parent(s): 2321bb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -13,6 +13,7 @@ def find(name, path):
13
 
14
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
15
  model.conf = 0.38
 
16
  #, force_reload=True
17
  def detect(inp):
18
  #g = (size / max(inp.size)) # gain
13
 
14
  model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
15
  model.conf = 0.38
16
+ model.hide_conf=True
17
  #, force_reload=True
18
  def detect(inp):
19
  #g = (size / max(inp.size)) # gain