dvieri commited on
Commit
e90973f
·
verified ·
1 Parent(s): 65f14a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def verify(image, model, person):
89
  if face is not None:
90
  if model == "Siamese":
91
  siamese = SiameseNetwork()
92
- siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'), map_location=torch.device('cpu'))
93
  siamese.eval()
94
 
95
  face = preprocess_image_siamese(face)
 
89
  if face is not None:
90
  if model == "Siamese":
91
  siamese = SiameseNetwork()
92
+ siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'))
93
  siamese.eval()
94
 
95
  face = preprocess_image_siamese(face)