dvieri commited on
Commit
a7d65df
·
verified ·
1 Parent(s): c17a362

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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)