Spaces:
Running
Running
Update app.py
Browse files
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 |
|