Add flagging arguments

app.py

@@ -70,7 +70,7 @@ def predict_fn(input_img):
     )
     logger.info(f"probabilities: {probabilities}")
     decision_json = json.dumps(
-        {"is_base_body": float(probabilities[0][1])}
+        {"is_base_body": float(probabilities[0][1].round(3))}
     ).encode("utf-8")
     logger.info(f"decision_json: {decision_json}")
     return decision_json
@@ -81,6 +81,11 @@ iface = gr.Interface(
     inputs="image",
     outputs="text",
     allow_flagging="manual",
+    flagging_options=[
+        "probability error > 0.5",
+        "0.2 < probability error < 0.5",
+        "probability error < 0.2"
+    ],
     flagging_callback=hf_writer
 )
 iface.launch()