dvieri commited on
Commit
1cc26c0
·
verified ·
1 Parent(s): 2dc58f5

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'))
93
 
94
  face = preprocess_image_siamese(face)
95
 
 
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
 
94
  face = preprocess_image_siamese(face)
95