Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def classify_image(input):
|
|
51 |
input_image=faceResized,
|
52 |
reshape_transform=gradCam.reshape_transform_vit_huggingface)
|
53 |
result = pipe(face)
|
54 |
-
if result[0]["label"] == "pos" and result[0]["score"] > 0.9:
|
55 |
return result[0]["label"], result[0]["score"], str("Nice!"), face, dffImage, gradCamImage
|
56 |
elif result[0]["label"] == "pos" and result[0]["score"] > 0.95:
|
57 |
return result[0]["label"], result[0]["score"], str("WHOA!!!!"), face, dffImage, gradCamImage
|
|
|
51 |
input_image=faceResized,
|
52 |
reshape_transform=gradCam.reshape_transform_vit_huggingface)
|
53 |
result = pipe(face)
|
54 |
+
if result[0]["label"] == "pos" and result[0]["score"] > 0.9 and result[0]["score"] < 0.95:
|
55 |
return result[0]["label"], result[0]["score"], str("Nice!"), face, dffImage, gradCamImage
|
56 |
elif result[0]["label"] == "pos" and result[0]["score"] > 0.95:
|
57 |
return result[0]["label"], result[0]["score"], str("WHOA!!!!"), face, dffImage, gradCamImage
|