Hugging Face's logo

Spaces:
kili-technology
/
plastic_in_river
Runtime error

App Files Community
Theo Dullin commited on
Commit
0f825bb
•
1 Parent(s): 76f9b06

update to model v3

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ torch.hub.download_url_to_file('https://storage.googleapis.com/kili-datasets-pub
7
  torch.hub.download_url_to_file('https://storage.googleapis.com/kili-datasets-public/plastic-in-river/samples/ckze0btj10ejd0lyyfzm85k9u.jpg', 'bottles2.jpg')
8
 
9
  # Model
10
- model = torch.hub.load('PierreLeveau/yolov5', 'custom', 'https://storage.googleapis.com/kili-datasets-public/plastic-in-river/model/v1/best.pt') # force_reload=True to update
11
 
12
 
13
  def yolo(im, size=640):
 
7
  torch.hub.download_url_to_file('https://storage.googleapis.com/kili-datasets-public/plastic-in-river/samples/ckze0btj10ejd0lyyfzm85k9u.jpg', 'bottles2.jpg')
8
 
9
  # Model
10
+ model = torch.hub.load('PierreLeveau/yolov5', 'custom', 'https://storage.googleapis.com/kili-datasets-public/plastic-in-river/model/v3/best.pt') # force_reload=True to update
11
 
12
 
13
  def yolo(im, size=640):