Fix bug in confidence thresholding

sociolome/lome_webserver.py

@@ -59,7 +59,7 @@ def make_prediction(sentence, spacy_model, predictor, confidence_threshold):
             continue
         if frm.upper() == frm:
             continue
-        if fr_proba.max() != confidence_threshold:
+        if fr_proba.max() < confidence_threshold:
             continue
         
         arg_spans, arg_labels, label_probas = predictor.force_decode(tokens, parent_span=tgt, parent_label=frm)
@@ -70,7 +70,7 @@ def make_prediction(sentence, spacy_model, predictor, confidence_threshold):
                 "roles": [
                     {"boundary": bnd, "label": label}
                     for bnd, label, probas in zip(arg_spans, arg_labels, label_probas)
-                    if label != "Target" and max(probas) == confidence_threshold
+                    if label != "Target" and max(probas) < confidence_threshold
                 ]
             }