wealthcoders commited on
Commit
c7cfe72
·
verified ·
1 Parent(s): a080294

Update handler.py

Browse files
Files changed (1) hide show
  1. 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]