Update handler.py (#4)

- Update handler.py (b287914b99f064f2f6685aa636e1a5e66b63de50)


Co-authored-by: Lewis Tunstall <[email protected]>

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)