lewtun HF staff commited on
Commit
b287914
1 Parent(s): c1a6092

Update handler.py

Browse files
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)