Spaces:

Gradio-Blocks
/

are-you-wearing-a-mask

Runtime error

hlydecker commited on May 18, 2022

Commit

09a96ae

•

1 Parent(s): c040456

content: added new, better model!

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,10 +6,9 @@ import numpy as np
 from PIL import Image
 # Face masks
-model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/face_masks_full.pt")
-# Animals
-# model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/datasets_1000_41class.pt",force_reload=True)

 from PIL import Image
 # Face masks
+# TODO: Allow user selectable model?
+model = torch.hub.load('ultralytics/yolov5', 'custom', "model_weights/face_masks_v8.pt")