handle

handler.py

@@ -542,8 +542,15 @@ class EndpointHandler:
         print(f"EndpointHandler initialized with model_dir: {model_dir}")
     
     def __call__(self, payload):
-        """Main endpoint function - delegates to query function"""
-        return query(payload)
+        """Main endpoint function - handles Hugging Face payload format"""
+        # Hugging Face sends payload in "inputs" wrapper
+        if "inputs" in payload:
+            # Extract the actual payload from inputs wrapper
+            actual_payload = payload["inputs"]
+            return query(actual_payload)
+        else:
+            # Direct payload (for backward compatibility)
+            return query(payload)
     
     def health_check(self):
         """Health check endpoint"""