fixed stupid mistake
Browse files- 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
|
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
|