Hugging Face's logo

Spaces:
NiiCole
/
FireExtinguishers
Runtime error

App Files Community
NiiCole commited on
Commit
c07f37a
•
1 Parent(s): aac289b

Update app.py

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://github.com/ultralytics/yolov5/raw/master
7
  torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/bus.jpg', 'bus.jpg')
8
 
9
  # Model
10
- model_name ='model.pt' # force_reload=True to update
11
 
12
  if model_name:
13
  model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_name, force_reload=True)
 
7
  torch.hub.download_url_to_file('https://github.com/ultralytics/yolov5/raw/master/data/images/bus.jpg', 'bus.jpg')
8
 
9
  # Model
10
+ model_name ='best.pt' # force_reload=True to update
11
 
12
  if model_name:
13
  model = torch.hub.load('ultralytics/yolov5', 'custom', path=model_name, force_reload=True)