Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ def verify(image, model, person, validation_image=None):
|
|
118 |
with torch.no_grad():
|
119 |
output = siamese(face, validation_face)
|
120 |
probability = output.item()
|
121 |
-
pred = 1.0 if probability > 0.
|
122 |
|
123 |
if pred == 1:
|
124 |
st.write("Match")
|
|
|
118 |
with torch.no_grad():
|
119 |
output = siamese(face, validation_face)
|
120 |
probability = output.item()
|
121 |
+
pred = 1.0 if probability > 0.5 else 0.0
|
122 |
|
123 |
if pred == 1:
|
124 |
st.write("Match")
|