ongkn commited on
Commit
18a581e
·
1 Parent(s): dda45e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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