Hugging Face's logo

Spaces:
Gradio-Blocks
/
are-you-wearing-a-mask
Runtime error

App Files Community
1
hlydecker commited on
Commit
2dff14b
•
1 Parent(s): 4adfc36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from PIL import Image
7
 
8
  # Face masks
9
  # TODO: Allow user selectable model?
10
- model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/face_masks_v8.pt")
11
 
12
  def yolo(im, size=640):
13
  g = (size / max(im.size)) # gain
 
7
 
8
  # Face masks
9
  # TODO: Allow user selectable model?
10
+ model = torch.hub.load('ultralytics/yolov5:v6.2', 'custom', "model_weights/face_masks_v8.pt")
11
 
12
  def yolo(im, size=640):
13
  g = (size / max(im.size)) # gain