Update app.py

app.py

@@ -82,10 +82,6 @@ def verify(image, model, person):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_image:
         temp_image.write(image.read())
         temp_image_path = temp_image.name
-        
-    image = cv2.imread(temp_image_path)
-
-    face = get_face(image)
 
     if face is not None:
         if model == "Siamese":
@@ -93,6 +89,10 @@ def verify(image, model, person):
             siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'))
             siamese.eval()
 
+            image = Image.open(temp_image_path)
+
+            face = get_face(image)
+
             face = preprocess_image_siamese(face)
         
             with torch.no_grad():
@@ -111,6 +111,10 @@ def verify(image, model, person):
             with open(f'./pca_{person.lower()}.pkl', 'rb') as f:
                 pca = joblib.load(f)
 
+            image = cv2.imread(temp_image_path)
+        
+            face = get_face(image)
+
             face = preprocess_image_svm(face)
 
             hog = extract_hog_features(face)