hb-setosys commited on
Commit
42a3da3
·
verified ·
1 Parent(s): bb8939e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from ultralytics import YOLO
6
 
7
  # Load YOLOv8 model (pre-trained on COCO dataset)
8
- model = YOLO("yolov11x.pt") # Use "yolov8x.pt" for highest accuracy
9
 
10
  # Class label for trucks (COCO dataset)
11
  TRUCK_CLASS_ID = 7 # "truck" in COCO dataset
 
5
  from ultralytics import YOLO
6
 
7
  # Load YOLOv8 model (pre-trained on COCO dataset)
8
+ model = YOLO("yolov8x.pt") # Use "yolov8x.pt" for highest accuracy
9
 
10
  # Class label for trucks (COCO dataset)
11
  TRUCK_CLASS_ID = 7 # "truck" in COCO dataset