ndk0706 commited on
Commit
584c4fa
1 Parent(s): f71e9a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,10 +13,10 @@ title = "Excavator Detector"
13
  description = "An Yolov8(m) object detection computer vision model to detect images of excavators."
14
 
15
  excavator_path = Path('.')
16
- example_path = excavator_path / "Examples"
17
 
18
  # Get example filepaths in a list of lists
19
- example_list = [["Examples/" + example] for example in os.listdir(example_path)]
20
 
21
  def predict(img, conf_thres=0.35, iou_thres=0.5): # default : conf(0.25), iou(0.7)
22
  model = YOLO('./weights/excavator_only_best_v8m.pt')
 
13
  description = "An Yolov8(m) object detection computer vision model to detect images of excavators."
14
 
15
  excavator_path = Path('.')
16
+ example_path = excavator_path / "examples"
17
 
18
  # Get example filepaths in a list of lists
19
+ example_list = [["examples/" + example] for example in os.listdir(example_path)]
20
 
21
  def predict(img, conf_thres=0.35, iou_thres=0.5): # default : conf(0.25), iou(0.7)
22
  model = YOLO('./weights/excavator_only_best_v8m.pt')