Update handler.py

handler.py

@@ -12,15 +12,17 @@ import spacy
 from typing import Dict, Any
 
 class EndpointHandler:
-    def __init__(self, path=""):
+    def __init__(self, path="./"):
         self.nlp = spacy.load(path)
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         text = data.get("text", "")
         doc = self.nlp(text)
-        entities = [{"text": ent.text, "start": ent.start_char, "end": ent.end_char, "label": ent.label_} for ent in doc.ents]
-        return {"entities": entities}
-
+        entities = [
+            {"word": ent.text, "entity_group": ent.label_, "score": 1.0} 
+            for ent in doc.ents
+        ]
+        return entities
 """
 handler = EndpointHandler()
 data = {"text": "Apple is looking at buying U.K. startup for $1 billion"}