Update handler.py

handler.py

@@ -20,11 +20,10 @@ class EndpointHandler():
         self.pipe = self.pipe.to(device)
 
 
-    def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
         Args:
-            data (:obj:):
-                includes the input data and the parameters for the inference.
+            images (:obj:`string`)
         Return:
             A :obj:`dict`:. base64 encoded image
         """