Update app.py

app.py

@@ -51,13 +51,18 @@ def classify_image(input):
                                         input_image=faceResized,
                                         reshape_transform=gradCam.reshape_transform_vit_huggingface)
     result = pipe(face)
-    return result[0]["label"], result[0]["score"], face, dffImage, gradCamImage
+    if result[0]["label"] == "pos" and result[0]["score"] > 0.9:
+        return result[0]["label"], result[0]["score"], str("Nice!"), face, dffImage, gradCamImage
+    elif result[0]["label"] == "pos" and result[0]["score"] > 0.95
+        return result[0]["label"], result[0]["score"], str("WHOA!!!!"), face, dffImage, gradCamImage
+    else:
+        return result[0]["label"], result[0]["score"], "Indifferent", face, dffImage, gradCamImage
 
 
 iface = gr.Interface(
     fn=classify_image,
     inputs="image",
-    outputs=["text", "number", "image", "image", "image"],
+    outputs=["text", "number", "text", "image", "image", "image"],
     title="Attraction Classifier - subjective",
     description=f"Takes in a (224, 224) image and outputs an attraction class: {'pos', 'neg'}, along with a GradCam/DFF explanation. Face detection, cropping, and resizing are done internally. Uploaded images are not stored by us, but may be stored by HF. Refer to their privacy policy for details."
 )