nicklorch commited on
Commit
5c8b4a1
·
1 Parent(s): a4e73a2

fixed stupid mistake

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -21,7 +21,7 @@ class EndpointHandler():
21
  if isinstance(data, dict):
22
  print('data is a dict: ', data)
23
  inputs = data.pop("inputs", None)
24
- text_input = inputs.get('text',None)
25
  image_data = BytesIO(base64.b64decode(inputs['image'])) if 'image' in inputs else None
26
  else:
27
  # assuming its an image sent via binary
 
21
  if isinstance(data, dict):
22
  print('data is a dict: ', data)
23
  inputs = data.pop("inputs", None)
24
+ text_input = inputs["text"] if "text" in inputs else None
25
  image_data = BytesIO(base64.b64decode(inputs['image'])) if 'image' in inputs else None
26
  else:
27
  # assuming its an image sent via binary