tarasevicius commited on
Commit
24f982f
·
1 Parent(s): 887ab5c

feat: hide labels

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,7 +9,8 @@ model_path = hf_hub_download(repo_id='ethandavey/yoloV5-coursework-model', filen
9
 
10
  # Load the YOLOv5 model
11
  model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, source='github')
12
- model.line_thickness = 1 # Set the thickness of the bounding box lines
 
13
 
14
 
15
  # Print model class names
 
9
 
10
  # Load the YOLOv5 model
11
  model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_path, source='github')
12
+ model.line_thickness = 3 # Set the thickness of the bounding box lines
13
+ model.hide_labels = True
14
 
15
 
16
  # Print model class names