Update app.py

app.py

@@ -77,7 +77,7 @@ def get_face(img):
         return img[y1:y2, x1:x2]
     return None
 
-def verify(image, model, person, validation_image=None):
+def verify(image, model, person, validation_image=None, threshold=None):
     
     with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as temp_image:
         temp_image.write(image.read())
@@ -118,7 +118,7 @@ def verify(image, model, person, validation_image=None):
             with torch.no_grad():
                 output = siamese(face, validation_face)
                 probability = output.item()
-                pred = 1.0 if probability > 0.5 else 0.0
+                pred = 1.0 if probability > threshold else 0.0
         
             if pred == 1:
                 st.write("Match")
@@ -150,16 +150,22 @@ def verify(image, model, person, validation_image=None):
 
 def main():
     st.title("Face Verification")
+
+    person_dict = {
+        "Theo": 0.542,
+        "Deverel": 0.5,
+        "Justin": 0.5
+    }
     
     model = st.selectbox("Select Model", ["Siamese", "HOG-SVM"])
-    person = st.selectbox("Select Person", ["Theo", "Deverel", "Justin"])
+    person = st.selectbox("Select Person", person_dict.keys())
     if model == "Siamese":
         uploaded_image = st.file_uploader("Upload Validation Image (Siamese)", type=["jpg", "png"])
     enable = st.checkbox("Enable camera")
     captured_image = st.camera_input("Take a picture", disabled=not enable)
 
     if captured_image and model == "Siamese":
-        verify(captured_image, model, person, uploaded_image)
+        verify(captured_image, model, person, uploaded_image, person_dict.get(person))
     elif captured_image and model == "HOG-SVM":
         verify(captured_image, model, person)