jeff-RQ commited on
Commit
40bec71
·
1 Parent(s): 5cce52e

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -15,7 +15,7 @@ class EndpointHandler:
15
 
16
  def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
17
  # process input
18
- # image = data.pop("image", data)
19
  text = data.pop("text", data)
20
 
21
  image_string = base64.b64decode(data["image"])
 
15
 
16
  def __call__(self, data: Dict[str, Any]) -> Dict[str, str]:
17
  # process input
18
+ data = data.pop("inputs", data)
19
  text = data.pop("text", data)
20
 
21
  image_string = base64.b64decode(data["image"])