philschmid HF staff lewtun HF staff commited on
Commit
7edc82d
1 Parent(s): c1a6092

Update handler.py (#4)

Browse files

- Update handler.py (b287914b99f064f2f6685aa636e1a5e66b63de50)


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

Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -11,8 +11,8 @@ class EndpointHandler:
11
  def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
12
  """
13
  Args:
14
- data (:obj:):
15
- includes the deserialized image file as PIL.Image
16
  """
17
  # process input
18
  inputs = data.pop("inputs", data)
 
11
  def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
12
  """
13
  Args:
14
+ data (:dict:):
15
+ The payload with the text prompt and generation parameters.
16
  """
17
  # process input
18
  inputs = data.pop("inputs", data)