Hugging Face's logo

Spaces:
xiang-wuu
/
yolov5
Runtime error

App Files Community
glenn-jocher commited on
Commit
c8f0694
•
1 Parent(s): a814720

PyTorch Hub update

Browse files
Files changed (1) hide show
  1. hubconf.py +1 -1
hubconf.py CHANGED
@@ -5,7 +5,7 @@ Usage:
5
  model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80)
6
  """
7
 
8
- dependencies = ['torch', 'pyyaml']
9
  import torch
10
 
11
  from models.yolo import Model
 
5
  model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80)
6
  """
7
 
8
+ dependencies = ['torch', 'yaml']
9
  import torch
10
 
11
  from models.yolo import Model