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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -90,6 +90,7 @@ def verify(image, model, person):
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
 
 
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)
96