Upload 2 files

handler.py

@@ -0,0 +1,25 @@
+from typing import Dict, List, Any
+import spacy
+
+
+class EndpointHandler:
+    def __init__(self, path=""):
+        self.nlp = spacy.load(path)
+
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        """
+            Your mom.
+        """
+
+        # Get inputs
+        text_input = data.pop("input", "")
+
+        # Perform regular model task
+        result = self.nlp(text_input)
+
+        # Return the result
+        entities = []
+        for ent in result.ents:
+            entities.append({"text": ent.text, "label": ent.label_, "id": ent.ent_id_})
+
+        return entities

requirements.txt

@@ -0,0 +1 @@
+spacy