bugfix: entities names without pos_ prefix.

app.py

@@ -296,7 +296,7 @@ def generate_answer(text):
            entities_formatted[f"pred_{feature_name}"] = extract_valid_string_left_dotted(entities_formatted[f"pred_{feature_name}"], text, entities_formatted[f"pos_{feature_name}"])
 
     # Clean Entities
-    entities_names = list({c.replace("pred_", "").replace("prob_", "") for c in list(entities_formatted.keys())})
+    entities_names = list({c.replace("pred_", "").replace("prob_", "").replace("pos_", "") for c in list(entities_formatted.keys())})
     entities_cleaned = dict()
     for feature_name in entities_names:
         entity_prediction_cleaned = clean_prediction(entities_formatted, feature_name, threshols_dict, clean_functions_dict)