Spaces:

mahmoud669
/

face-celebs-scrub

Sleeping

mahmoud669 commited on Jun 23, 2024

Commit

1ac03ab

verified ·

1 Parent(s): d1b1c42

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,7 +40,7 @@ reversed_map = {
 }
 model = timm.create_model("rexnet_150", pretrained = True, num_classes = 17)
-model.load_state_dict(torch.load('face_best_model.pth', map_location=torch.device('cpu')))
 model.eval()
 # Title of the app

 }
 model = timm.create_model("rexnet_150", pretrained = True, num_classes = 17)
+model.load_state_dict(torch.load('faces_best_model.pth', map_location=torch.device('cpu')))
 model.eval()
 # Title of the app