Update handler.py

handler.py

@@ -28,15 +28,16 @@ class EndpointHandler():
         #self.pipe = self.pipe.to(device)
 
 
-    def __call__(self, *args, **kwargs) -> List[Dict[str, Any]]:
+    def __call__(self, data) -> List[Dict[str, Any]]:
         """
         Args:
             image (:obj:`string`)
         Return:
             A :obj:`dict`:. base64 encoded image
         """
-        logger.info('args received %s', args)
-        logger.info('kwargs received %s', kwargs)
+        logger.info('data received %s', data)
+        inputs = data.get("inputs")
+        logger.info('inputs received %s', inputs)
 
         # postprocess the prediction
         return {"image": "nothing"}