Sa-m commited on
Commit
347c196
1 Parent(s): 45cf92d

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', force_reload=True)
 
23
  def detect(inp):
24
  #g = (size / max(inp.size)) # gain
25
  #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize
 
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
26
  #im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize