Update generic_ner.py

generic_ner.py

@@ -106,11 +106,12 @@ def get_entities(tokens, tags, confidences, text):
                 entities.append(
                     {
                         "entity": original_label,
-                        "score": int(np.average(confidences[idx : idx + len(subtree)])),
+                        "score": np.average(confidences[idx : idx + len(subtree)]),
                         "index": idx,
                         "word": original_string,
                         "start": entity_start_position,
                         "end": entity_end_position,
+                        "text": text
                     }
                 )
                 print(text[entity_start_position:entity_end_position], '-------', original_string)