Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -111,6 +111,9 @@ def verify(image, model, person, validation_image=None):
|
|
111 |
|
112 |
face = preprocess_image_siamese(temp_face_path)
|
113 |
validation_face = preprocess_image_siamese(validation_temp_image_path)
|
|
|
|
|
|
|
114 |
|
115 |
with torch.no_grad():
|
116 |
output = siamese(face, validation_face)
|
|
|
111 |
|
112 |
face = preprocess_image_siamese(temp_face_path)
|
113 |
validation_face = preprocess_image_siamese(validation_temp_image_path)
|
114 |
+
|
115 |
+
face = face.unsqueeze(0)
|
116 |
+
validation_face = validation_face.unsqueeze(0)
|
117 |
|
118 |
with torch.no_grad():
|
119 |
output = siamese(face, validation_face)
|