Update handler.py

handler.py

@@ -11,8 +11,8 @@ class EndpointHandler:
     def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
         """
         Args:
-            data (:obj:):
-                includes the deserialized image file as PIL.Image
+            data (:dict:):
+                The payload with the text prompt and generation parameters.
         """
         # process input
         inputs = data.pop("inputs", data)