Hugging Face's logo

Spaces:
GordonM
/
BandiCount
Runtime error

App Files Community
GordonM commited on
Commit
66ec0c6
·
1 Parent(s): 3c44234

try older version yolov5

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  from PIL import Image
6
 
7
  # Load model weights
8
- model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/datasets_1000_41class.pt")
9
 
10
  # Define a yolo prediction function
11
  def yolo(im, size=640):
 
5
  from PIL import Image
6
 
7
  # Load model weights
8
+ model = torch.hub.load('ultralytics/yolov5:v6.2', 'custom', "model_weights/datasets_1000_41class.pt")
9
 
10
  # Define a yolo prediction function
11
  def yolo(im, size=640):