Add requirements.txt for endpoint dependencies

handler.py

@@ -1,14 +1,15 @@
 from gliner import GLiNER
-from typing import Dict
+from typing import Dict, List, Any
 
-class EndpointHandler:
+
+class EndpointHandler():
     def __init__(self, path=""):
-        self.model = GLiNER.from_pretrained(path)
-
-    def __call__(self, data: Dict[str, str]) -> Dict:
-        text = data.get("text", "")
-        labels = data.get("labels", [])
-        if not text or not labels:
-            return {"error": "Missing 'text' or 'labels'"}
-        entities = self.model.predict_entities(text, labels)
-        return {"entities": entities}
+        self.model = GLiNER.from_pretrained("Mbiot/gliner-custom")
+
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        # get inputs
+        inputs = data.pop("inputs", data)
+        labels = data.pop("labels", None)
+
+        prediction = self.model.predict_entities(inputs, labels)
+        return prediction