Hugging Face's logo

Spaces:
mahmoud669
/
face-celebs-scrub
Sleeping

App Files Community
mahmoud669 commited on
Commit
1372c37
•
1 Parent(s): 92abc99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ reversed_map = {
38
  16: 'Will Smith'
39
  }
40
 
41
- model = timm.create_model("rexnet_150", pretrained = True, num_classes = len(classes))
42
  model.load_state_dict(torch.load('faces_best_model.pth', map_location=torch.device('cpu')))
43
  model.eval()
44
 
 
38
  16: 'Will Smith'
39
  }
40
 
41
+ model = timm.create_model("rexnet_150", pretrained = True, num_classes = 17)
42
  model.load_state_dict(torch.load('faces_best_model.pth', map_location=torch.device('cpu')))
43
  model.eval()
44