kevinconka commited on
Commit
831bf7d
·
verified ·
1 Parent(s): ab899b5

Update model_yolov5.py

Browse files
Files changed (1) hide show
  1. model_yolov5.py +1 -0
model_yolov5.py CHANGED
@@ -9,6 +9,7 @@ def load_model(model_path, img_size=640):
9
  """Load model from HuggingFace Hub."""
10
  model = yolov5.load(model_path, hf_token=get_token())
11
  model.img_size = img_size # add img_size attribute
 
12
  return model
13
 
14
 
 
9
  """Load model from HuggingFace Hub."""
10
  model = yolov5.load(model_path, hf_token=get_token())
11
  model.img_size = img_size # add img_size attribute
12
+ model.conf = 0.15
13
  return model
14
 
15