Update app.py

app.py

@@ -83,7 +83,6 @@ def verify(image, model, person):
         temp_image.write(image.read())
         temp_image_path = temp_image.name
 
-    if face is not None:
         if model == "Siamese":
             siamese = SiameseNetwork()
             siamese.load_state_dict(torch.load(f'siamese_{person.lower()}.pth'))
@@ -93,17 +92,20 @@ def verify(image, model, person):
 
             face = get_face(image)
 
-            face = preprocess_image_siamese(face)
-        
-            with torch.no_grad():
-                output = model(face)
-                probability = output.item()
-                pred = 1.0 if probability > 0.7 else 0.0
-        
-            if pred == 1:
-                st.write("Match")
+            if face is not None:
+                face = preprocess_image_siamese(face)
+            
+                with torch.no_grad():
+                    output = model(face)
+                    probability = output.item()
+                    pred = 1.0 if probability > 0.7 else 0.0
+            
+                if pred == 1:
+                    st.write("Match")
+                else:
+                    st.write("Not Match")
             else:
-                st.write("Not Match")
+                st.write("Face not detected")
         
         elif model == "HOG-SVM":
             with open(f'./svm_{person.lower()}.pkl', 'rb') as f:
@@ -115,20 +117,21 @@ def verify(image, model, person):
         
             face = get_face(image)
 
-            face = preprocess_image_svm(face)
-
-            hog = extract_hog_features(face)
-
-            hog_pca = pca.transform([hog])
-
-            pred = svm.predict(hog_pca)
-
-            if pred == 1:
-                st.write("Match")
+            if face is not None:
+                face = preprocess_image_svm(face)
+    
+                hog = extract_hog_features(face)
+    
+                hog_pca = pca.transform([hog])
+    
+                pred = svm.predict(hog_pca)
+    
+                if pred == 1:
+                    st.write("Match")
+                else:
+                    st.write("Not Match")
             else:
-                st.write("Not Match")
-    else:
-        st.write("Face not detected")
+                st.write("Face not detected")
 
 def main():
     st.title("Face Verification")