Spaces:
Sleeping
Sleeping
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)
|