earlalvarado-pi
/

en_core_web_sm

Token Classification

Inference Endpoints

Model card Files Files and versions Community

earlalvarado-pi commited on Mar 26

Commit

c7a97db

•

1 Parent(s): 8d90324

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -19,7 +19,7 @@ class EndpointHandler:
         self.nlp = spacy.load(path)
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
-        text = data.get("text", "")
         doc = self.nlp(text)
         entities = [
             {"word": ent.text, "entity_group": ent.label_, "score": 1.0}

         self.nlp = spacy.load(path)
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
+        text = data.get("inputs", "")
         doc = self.nlp(text)
         entities = [
             {"word": ent.text, "entity_group": ent.label_, "score": 1.0}