Update handler.py

handler.py

@@ -27,9 +27,7 @@ class EndpointHandler:
         formatted_results = []
         for entity in entities:
             formatted_entity = {
-                "word": entity["text"],
-                "entity_group": entity["label"],  # Assuming entity["label"] contains the label
-                "score": entity.get("score", 1.0)  # Default to 1.0 if score not provided
+               entity["label"]: entity["text"],
             }
             formatted_results.append(formatted_entity)