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
|