Update handler.py

handler.py

@@ -46,6 +46,12 @@ class EndpointHandler:
         # ✅ Compatibility layer (does NOT break anything)
         payload = data.get("inputs", data)
 
+        # ✅ Robust guard: prevents KeyError and hard crashes
+        if not isinstance(payload, dict) or "image" not in payload:
+            return {
+                "error": "Missing required key: 'image'. Expected base64-encoded image."
+            }
+
         image_b64 = payload["image"]
         conf = float(payload.get("conf", 0.25))
         iou = float(payload.get("iou", 0.45))