dvieri commited on
Commit
5266c97
·
verified ·
1 Parent(s): c83f270

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -50,12 +50,12 @@ class SiameseNetwork(nn.Module):
50
  x = self.sigmoid(x)
51
  return x
52
 
53
- def preprocess_image_siamese(img):
54
  transform = transforms.Compose([
55
  transforms.Resize((224, 224)),
56
  transforms.ToTensor()
57
  ])
58
- img = Image.open(img).convert("RGB")
59
  return transform(img)
60
 
61
  def preprocess_image_svm(img):
@@ -95,10 +95,8 @@ def verify(image, model, person):
95
  siamese = SiameseNetwork()
96
  siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'))
97
  siamese.eval()
98
-
99
- face = Image.open(temp_face_path)
100
 
101
- face = preprocess_image_siamese(face)
102
 
103
  with torch.no_grad():
104
  output = model(face)
 
50
  x = self.sigmoid(x)
51
  return x
52
 
53
+ def preprocess_image_siamese(temp_face_path):
54
  transform = transforms.Compose([
55
  transforms.Resize((224, 224)),
56
  transforms.ToTensor()
57
  ])
58
+ img = Image.open(temp_face_path).convert("RGB")
59
  return transform(img)
60
 
61
  def preprocess_image_svm(img):
 
95
  siamese = SiameseNetwork()
96
  siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'))
97
  siamese.eval()
 
 
98
 
99
+ face = preprocess_image_siamese(temp_face_path)
100
 
101
  with torch.no_grad():
102
  output = model(face)