Update handler.py
Browse files- handler.py +3 -0
handler.py
CHANGED
|
@@ -43,6 +43,9 @@ class EndpointHandler:
|
|
| 43 |
try:
|
| 44 |
inputs = data.get("inputs")
|
| 45 |
base64_string = inputs["base64"]
|
|
|
|
|
|
|
|
|
|
| 46 |
# Remove data URL prefix if present
|
| 47 |
if ',' in base64_string:
|
| 48 |
base64_string = base64_string.split(',')[1]
|
|
|
|
| 43 |
try:
|
| 44 |
inputs = data.get("inputs")
|
| 45 |
base64_string = inputs["base64"]
|
| 46 |
+
|
| 47 |
+
if not base64_string:
|
| 48 |
+
return {"error": "No base64 string provided"}
|
| 49 |
# Remove data URL prefix if present
|
| 50 |
if ',' in base64_string:
|
| 51 |
base64_string = base64_string.split(',')[1]
|