Update flagging arguments and add decision

app.py

@@ -69,8 +69,17 @@ def predict_fn(input_img):
         cosine_simlarities.reshape(1, -1)
     )
     logger.info(f"probabilities: {probabilities}")
+    result_probabilty = float(probabilities[0][1].round(3))
+    # get decision string
+    if result_probabilty > 0.95:
+        decision = "AUTO ACCEPT"
+    elif result_probabilty < 0.4:
+        decision = "AUTO REJECT"
+    else:
+        decision = "MODERATION"
+    logger.info(f"decision: {decision}")
     decision_json = json.dumps(
-        {"is_base_body": float(probabilities[0][1].round(3))}
+        {"is_base_body": result_probabilty, "decision": decision}
     ).encode("utf-8")
     logger.info(f"decision_json: {decision_json}")
     return decision_json
@@ -86,15 +95,14 @@ iface = gr.Interface(
     probability < 0.4, the image can be automatically REJECTED as NOT as base
     body. All other cases will be submitted for moderation.
 
-    To minimize noise, please flag the input only if you think there is an error
-    in the probability returned by the model and it is off by at least 0.2
+    Please flag if you think the decision is wrong.
     """,
 
     allow_flagging="manual",
     flagging_options=[
-        "probability error > 0.5",
-        "0.2 < probability error < 0.5",
-        "probability error < 0.2"
+        ": decision hould be accept",
+        ": decision hould be reject",
+        ": decision hould be moderation"
     ],
     flagging_callback=hf_writer
 )