Update handler.py

handler.py

@@ -2,7 +2,7 @@ from typing import Dict, List, Any
 from gliner import GLiNER
 
 class EndpointHandler:
-    def __init__(self, path=""):
+    def __init__(self, path="."):
         # Initialize the GLiNER model
         self.model = GLiNER.from_pretrained("urchade/gliner_multi-v2.1")
 
@@ -19,7 +19,8 @@ class EndpointHandler:
         # Get inputs and labels
         inputs = data.get("inputs", "")
         labels = ["party", "document title"]
-
+        print('path', path)
+        print('labels',labels)
         # Predict entities using GLiNER
         entities = self.model.predict_entities(inputs, labels)
 
@@ -29,6 +30,7 @@ class EndpointHandler:
             formatted_entity = {
                entity["label"]: entity["text"],
             }
+            print(formatted_entity)
             formatted_results.append(formatted_entity)
 
         return formatted_results