new output format

README.md

@@ -1,3 +1,14 @@
 ---
+language: en
 license: mit
+tags:
+- spacy
+- token-classification
+widgets:
+- example_title: "Example 1"
+  text: "My name is Sarah and I live in London."
+- example_title: "Example 2"
+  text: "Who is John Doe?"
 ---
+
+English pipeline optimized for CPU. Components: ner.

handler.py

@@ -25,7 +25,7 @@ class EndpointHandler:
         print(f"EndpointHandler(path='{path}')")
         self._nlp: spacy.Language = _load_spacy_model(name="en_core_web_lg")
 
-    def __call__(self, data: dict[str, Any]) -> dict[str, Any]:
+    def __call__(self, data: dict[str, Any]) -> list[dict[str, Any]]:
         inputs: str = data.pop("inputs", "")
         if not inputs:
             return {}
@@ -38,12 +38,12 @@ class EndpointHandler:
                 continue
 
             entity: dict = {
-                "qid": None,
-                "entity": ent.label_,
-                "text": ent.text,
+                "entity_group": ent.label_,
+                "score": 1,
+                "word": ent.text,
                 "start": ent.start_char,
                 "end": ent.end_char}
 
             outputs.append(entity)
 
-        return {"outputs": outputs}
+        return outputs