Akbartus commited on
Commit
04e5ce6
·
1 Parent(s): f0c5c7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from PIL import Image
5
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/06/15/01/11/soccer-1457988_1280.jpg', 'soccer.jpg')
6
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/11/21/14/31/vw-bus-1845719_1280.jpg', 'bus.jpg')
7
  # Model
8
- model = torch.hub.load('ultralytics/yolov3', 'yolov3') # or yolov3-spp, yolov3-tiny, custom
9
  def yolo(im, size=1920):
10
  g = (size / max(im.size)) # gain
11
  im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize
 
5
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/06/15/01/11/soccer-1457988_1280.jpg', 'soccer.jpg')
6
  torch.hub.download_url_to_file('https://cdn.pixabay.com/photo/2016/11/21/14/31/vw-bus-1845719_1280.jpg', 'bus.jpg')
7
  # Model
8
+ model = torch.hub.load('ultralytics/yolov5', 'yolov5') # or yolov3-spp, yolov3-tiny, custom
9
  def yolo(im, size=1920):
10
  g = (size / max(im.size)) # gain
11
  im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize