ndk0706 commited on
Commit
f71e9a5
1 Parent(s): 47532b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ 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')
 
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')