Sa-m commited on
Commit
df9fea9
·
1 Parent(s): 347c196

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ def find(name, path):
19
  if name in files:
20
  return os.path.join(root, name)
21
 
22
- model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt',conf_thres=0.45)
 
23
  #, force_reload=True
24
  def detect(inp):
25
  #g = (size / max(inp.size)) # gain
 
19
  if name in files:
20
  return os.path.join(root, name)
21
 
22
+ model = torch.hub.load('ultralytics/yolov5', 'custom', path='Content/best.pt')
23
+ model.conf(0.45)
24
  #, force_reload=True
25
  def detect(inp):
26
  #g = (size / max(inp.size)) # gain