Blessing commited on
Commit
dd0ae6d
1 Parent(s): 1eb6e92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ for i, url in enumerate(file_urls):
33
  )
34
 
35
 
36
- model = YOLO('best.pt')
37
  path = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg'], ['image_3.jpg']]
38
 
39
  # path = [['IMG_7612.JPG'], ['IMG_7678.JPG'], ['all_33.jpg'], ['all_80.jpg'],
 
33
  )
34
 
35
 
36
+ model = YOLO('best.onnx', task='detect')
37
  path = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg'], ['image_3.jpg']]
38
 
39
  # path = [['IMG_7612.JPG'], ['IMG_7678.JPG'], ['all_33.jpg'], ['all_80.jpg'],