Hugging Face's logo

Spaces:
StevenLimcorn
/
mask-detection
Sleeping

App Files Community
StevenLimcorn commited on
Commit
c056859
•
1 Parent(s): 44b4267

updated

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch.nn.functional as F
4
  import torch
5
  from torchvision import transforms
6
 
7
- model = torch.load("/content/drive/MyDrive/Mask Detection/model.pth", map_location=torch.device("cpu"))
8
  IMG_SIZE = 224
9
  MASK_LABEL = ["Mask worn properly.", "Mask not worn properly: nose out", "Mask not worn properly: chin and nose out", "Didn't wear mask."]
10
 
 
4
  import torch
5
  from torchvision import transforms
6
 
7
+ model = torch.load("./model.pth", map_location=torch.device("cpu"))
8
  IMG_SIZE = 224
9
  MASK_LABEL = ["Mask worn properly.", "Mask not worn properly: nose out", "Mask not worn properly: chin and nose out", "Didn't wear mask."]
10